casa6

from __future__ import absolute_import

from __future__ import print_function

import os

import shutil

​

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 casatools import ctsys

    from casatasks import bandpass

else:

    #import testhelper as th

    from __main__ import default

    from tasks import bandpass

    from taskinit import *

​

from casatestutils import testhelper as th

import unittest

​

​

''' Python unit tests for the bandpass task

​

These tests will only verify if the bandpass calibration

tables created for an MS and an MMS agree. These are

not full unit tests for the bandpass task.

'''

​

if is_CASA6:

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

else:

    datapath = os.environ.get('CASAPATH').split()[0] +\

               '/casatestdata/unittest/bandpass/'

​

# 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'))+'/bandpass/'

    if os.path.isdir(DATADIR):

        testmms = True

        datapath = DATADIR

    else:

        print('WARN: directory '+DATADIR+' does not exist')

​

print('bandpass tests will use data from '+datapath)

​

# Base class which defines setUp functions

# for importing different data sets

class test_base(unittest.TestCase):

​

    def cleanUp(self):

        shutil.rmtree(self.msfile, ignore_errors=True)

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

    def setUp_ngc5921(self):

        # Input names

        prefix = 'ngc5921'

        self.msfile = prefix + '.ms'

#        if testmms:

#            self.msfile = prefix + '.mms'