casa6

from __future__ import absolute_import

from __future__ import print_function

import os

import shutil

import unittest

​

from casatasks.private.casa_transition import is_CASA6

if is_CASA6:

    ### for testhelper import

    import sys

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

    #import testhelper as th

    from casatasks.private.parallel.parallel_data_helper import ParallelDataHelper

​

    from casatasks import hanningsmooth, mstransform, partition, cvel, split, clearcal

    from casatools import ctsys, ms

​

    # default is not used in CASAtasks

    def default(atask):

        pass

​

    # Path for data

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

else:

    from __main__ import default

    from tasks import hanningsmooth, mstransform, partition, cvel, split, clearcal

    from taskinit import mstool as ms

    #import testhelper as th

    from parallel.parallel_data_helper import ParallelDataHelper

​

    # Path for data

    datapath = os.environ.get('CASAPATH').split()[0] + '/casatestdata/unittest/hanningsmooth/'

​

from casatestutils import testhelper as th

​

'''

functional tests for task hanningsmooth

'''

​

# Pick up alternative data directory to run tests on MMSs

testmms = False

if 'TEST_DATADIR' in os.environ:   

    DATADIR = str(os.environ.get('TEST_DATADIR'))+'/hanningsmooth/'

    if os.path.isdir(DATADIR):

        testmms = True

        datapath = DATADIR

​

print('hanningsmooth tests will use data from %s' % datapath)

​

class test_base(unittest.TestCase):

​

    def setUp_ngc5921(self):

        # only DATA column

        self.msfile = 'ngc5921_ut.ms'

        if testmms:

            self.msfile = 'ngc5921_ut.mms'

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

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

        default(hanningsmooth)

    def setUp_almams(self):

        # MS with DATA and CORRECTED_DATA

        self.msfile = 'ALMA-data-mst-science-testing-CAS-5013-one-baseline-one-timestamp.ms'

        if testmms:

            self.msfile = 'ALMA-data-mst-science-testing-CAS-5013-one-baseline-one-timestamp.mms'

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

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

        default(hanningsmooth)

​

    def createMMS(self, msfile, column='data', axis='auto',scans='',spws=''):

        '''Create MMSs for tests with input MMS'''

        prefix = msfile.rstrip('.ms')

        if not os.path.exists(msfile):

            os.system('cp -RL '+os.path.join(datapath,msfile)+' '+ msfile)

        # Create an MMS for the tests

        self.testmms = prefix + ".test.mms"

        default(partition)

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

            os.system("rm -rf " + self.testmms)

        print("................. Creating test MMS ..................")

        partition(vis=msfile, outputvis=self.testmms, datacolumn=column,

                    createmms=True,separationaxis=axis, scan=scans, spw=spws)

​

​