casa6

from __future__ import absolute_import

import os

import sys

import shutil

import unittest

​

from casatasks.private.casa_transition import is_CASA6

if is_CASA6:

    from casatools import ctsys, table

    from casatasks import smoothcal

​

    _tb = table()

else:

    from __main__ import default

    from tasks import *

    from taskinit import *

    from casa_stack_manip import stack_frame_find

    casa_stack_rethrow = stack_frame_find().get('__rethrow_casa_exceptions', False)

​

    _tb = tbtool()

​

'''

Unit tests for task smoothcal. It tests the following parameters:

    vis:           wrong and correct values

    tablein:       wrong and correct values

    caltable:      existence of output

    field:         wrong field value; non-default value

    smmothtype:    unsupported value; non-default value

    smoothtime:    unsupported value; non-default values

​

    Other tests: check the values of column smoothed GAIN against reference.

'''

class smoothcal_test(unittest.TestCase):

​

    # Input and output names

    msfile = 'ngc1333_ut.ms'

    gcal = 'ngc1333_ut_nct.gcal'   # New format caltables

    ref = 'ngc1333_ut_nct.ref'

    res = None

    out = 'smoothcal_test'

​

    def setUp(self):

        self.res = None

        if not is_CASA6:

            default(smoothcal)

        if is_CASA6:

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

        else:

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

​

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

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

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

    def tearDown(self):

        if (os.path.exists(self.msfile)):

            os.system('rm -rf ' + self.msfile)

        if (os.path.exists(self.gcal)):

            os.system('rm -rf ' + self.gcal)

        if (os.path.exists(self.ref)):

            os.system('rm -rf ' + self.ref)

        if (os.path.exists(self.out)):

            os.system('rm -rf ' + self.out)

    def getvarcol(self,table,colname):

        '''Return the requested column'''

        _tb.open(table)

        col = _tb.getvarcol(colname)

        _tb.close()

        return col

​

    def test0(self):

        '''Test 0: Missing input table caught by parameter checking (exception thrown)

        '''

        # CASA5 returns False (depending on __rethrow_casa_exceptions)

        try:

            OK = False

            self.res = smoothcal()

            if not is_CASA6:

                OK = not self.res

        except:

            if is_CASA6 or casa_stack_rethrow:

                OK = True

        self.assertTrue(OK)

​

    def test1(self):

        """Test 1: Wrong input MS caught by parameter checking (exception thrown)

        """

        msfile = 'badmsfile'

        # CASA5 returns False (depending on __rethrow_casa_exceptions)

        try: