casa6

import os

import unittest

​

import numpy

​

try:

    # CASA 6

    from casatools import calanalysis, ctsys

    ca = calanalysis()

    ca_datapath = ctsys.resolve('unittest/calanalysis/')

except ImportError:

    # CASA 5

    import casac

    ca = casac.casac.calanalysis()

    ca_datapath = os.environ.get('CASAPATH').split()[0]

    ca_datapath = os.path.join(ca_datapath, 'casatestdata/unittest/calanalysis/')

​

​

class calanalysis_tsys1_base(unittest.TestCase):

​

    """A Tsys calibration table is used in these tests."""

​

    calName = 'uid___A002_X30a93d_X43e.ms.tsys.s3.tbl'

    msName = 'uid___A002_X30a93d_X43e.ms'

    parType = 'Float'

    polBasis = 'U'

    visCal = 'B TSYS'

​

    fieldName = ['J2253+161; 3c454.3', 'Callisto',

                 'B0007+106; J0010+109', 'GRB021004']

    fieldNumber = ['0', '1', '2', '3']

    numField = len(fieldNumber)

​

    antennaName = ['DA41', 'DA42', 'DA43', 'DV02', 'DV03', 'DV05',

                   'DV07', 'DV10', 'DV11', 'DV12', 'DV13', 'DV14', 'PM02', 'PM03']

    antennaNumber = ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9',

                     '10', '11', '12', '13']

    numAntenna = len(antennaNumber)

​

    antenna1Name = antennaName

    antenna1Number = antennaNumber

    numAntenna1 = numAntenna

​

    antenna2Name = ['NoName']

    antenna2Number = ['-1']

    numAntenna2 = 1

​

    feed = ['1', '2']

    numFeed = len(feed)

​

    spwName = ['', '', '', '', '', '', '', '', '', '', '', '', '', '',

               '', '', '', '', '', '', '', '', '', '', '', '', '', '', '', '', '']

    spwNumber = ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9',

                 '10', '11', '12', '13', '14', '15', '16', '17', '18', '19', '20',

                 '21', '22', '23', '24', '25', '26', '27', '28', '29', '30']

    numSPW = len(spwNumber)

​

    numChannel = numpy.array([4, 128, 1, 128, 1, 128, 1, 128, 1, 128, 1,

                              128, 1, 128, 1, 128, 1, 4, 4, 4, 4, 4, 4, 4, 4, 4,

                              4, 4, 4, 4, 4], dtype=numpy.int32)

​

    time = numpy.array([4827167647.3920002, 4827167780.5439997,

                        4827168227.2320004, 4827168353.7600002, 4827168887.7600002,

                        4827169009.632, 4827169543.6800003, 4827169665.6960001,

                        4827170204.3520002, 4827170332.4160004, 4827170866.5600004,

                        4827170989.5360003, 4827171523.776, 4827171647.2799997])

    numTime = len(time)

​

    def setUp(self):

        os.system('cp -RH {0} {1}'.format(os.path.join(ca_datapath, self.calName),

                                          self.calName))

​

        return ca.open(self.calName)

​

    def tearDown(self):

        os.system('rm -rf {0}'.format(self.calName))

        return ca.close()

​

​

class calanalysis_tsys1_introspective(calanalysis_tsys1_base):

    """ This is a very simple unit test for introspective methods. """

​

    def test_introspective(self):

​

        """Test of introspective member functions"""

​

        self.assertEqual(os.path.split(ca.calname())[1],

                         self.calName)

        self.assertEqual(os.path.split(ca.msname())[1],

                         self.msName)

        self.assertEqual(ca.partype(), self.parType)