casa6

import shutil

import unittest

import os

import filecmp

import pprint

​

try:

    # CASA 6

    from casatools import agentflagger, ctsys

    ag_datapath = ctsys.resolve('unittest/agentflagger/')

except ImportError:

    # CASA 5

    from taskinit import aftool as agentflagger

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

    ag_datapath = os.path.join(ag_datapath, 'casatestdata/unittest/agentflagger/')

​

​

class test_base(unittest.TestCase):

​

    def setUp_bpass_case(self):

        self.vis = "cal.fewscans.bpass"

​

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

            print("The CalTable is already around, just unflag")

        else:

            print("Moving data...")

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

​

        os.system('rm -rf ' + self.vis + '.flagversions')

        self.unflag_table()

​

​

    def setUp_tsys_case(self):

        self.vis = "X7ef.tsys"

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

            print("The CalTable is already around, just unflag")

        else:

            print("Moving data...")

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

​

        os.system('rm -rf ' + self.vis + '.flagversions')

​

        self.unflag_table()

​

​

    def setUp_4Ants(self):

        self.vis = "Four_ants_3C286.ms"

​

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

            print("The MS is already around, just unflag")

        else:

            print("Moving data...")

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

​

        os.system('rm -rf ' + self.vis + '.flagversions')

​

        self.unflag_table()

​

​

    def setUp_CAS_4052(self):