casa6

from __future__ import absolute_import

from __future__ import print_function

import os

import sys

import shutil

import re

import numpy

import math

import unittest

​

from casatasks.private.casa_transition import is_CASA6

if is_CASA6:

    from casatools import ctsys

    from casatools import table as tbtool

    from casatools import ms as mstool

    from casatasks import sdgaincal, mstransform, sdcal

    from casatasks.private import sdutil

​

    ### for testhelper import

    sys.path.append(os.path.abspath(os.path.dirname(__file__)))

​

    # this distinction doesn't exist in casatasks

    mstransform_cli = mstransform

    sdcal_cli = sdcal

else:

    from __main__ import default

    from taskinit import tbtool, mstool

    from sdgaincal import sdgaincal

    from mstransform_cli import mstransform_cli

    from sdcal_cli import sdcal_cli

    import sdutil

​

​

class sdgaincal_test_base(unittest.TestCase):

    """

    Base class for sdgainal unit tests.

​

    This class defines attributes and methods common to test cases

    """

    if is_CASA6:

        datapath=ctsys.resolve('unittest/sdgaincal/')

    else:

        datapath=os.path.join(os.environ.get('CASAPATH').split()[0],'casatestdata/unittest/sdgaincal/')

    def __copy_from_datapath(self, filename):

        if os.path.exists(filename):

            shutil.rmtree(filename)

        shutil.copytree(os.path.join(self.datapath, filename), filename)

​

    def setUp(self):

        self.__copy_from_datapath(self.infile)

​

        if hasattr(self, 'reffile'):

            self.__copy_from_datapath(self.reffile)

​

        if not is_CASA6:

            default(sdgaincal)

​

    def tearDown(self):

        to_be_removed = [self.infile, self.outfile]

        if hasattr(self, 'reffile'):

            to_be_removed.append(self.reffile)

        for f in to_be_removed: