casa6

import os

import shutil

import time

import unittest

​

from casatestutils import testhelper as th

​

from casatasks.private.casa_transition import is_CASA6

if is_CASA6:

    from casatasks import listobs, listpartition, flagdata, flagmanager, partition, setjy, split

    from casatools import ctsys, msmetadata, ms, agentflagger, table

    from casatasks.private import partitionhelper as ph   ##### <----<<< this dependency should be removed

​

    msmdt_local = msmetadata()

    msmdt_local2 = msmetadata()

    mst_local = ms()

    aft_local = agentflagger()

    tbt_local = table()

​

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

else:

    from tasks import partition, flagdata, flagmanager, split, setjy, listpartition, listobs

    from taskinit import msmdtool, mstool, aftool, tbtool

​

    from parallel.parallel_task_helper import ParallelTaskHelper

    import partitionhelper as ph

​

    msmdt_local = msmdtool()

    msmdt_local2 = msmdtool()

    mst_local = mstool()

    aft_local = aftool()

    tbt_local = tbtool()

​

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

                       "/casatestdata/unittest/partition/"

​

from casatestutils import testhelper as th

''' Unit Tests for task partition'''

​

# jagonzal (CAS-4287): Add a cluster-less mode to by-pass parallel processing for MMSs as requested 

if 'BYPASS_PARALLEL_PROCESSING' in os.environ:

    ParallelTaskHelper.bypassParallelProcessing(1)

​

class test_base(unittest.TestCase):

    def setUp_ngc4826(self):

        # Input MS contain 4 obsID, 11 scans and 4 spws

        self.prefix = 'ngc4826'

        self.msfile = self.prefix + '.ms'  

        # Output files                      

        self.mmsfile = self.prefix + '.mms'

        fpath = os.path.join(datapath,self.msfile)

#         if os.path.lexists(fpath):        

#             shutil.copytree(fpath, self.msfile, symlinks=True)

#         else:

#             self.fail('Data does not exist -> '+fpath)

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

            shutil.copytree(fpath, self.msfile, symlinks=True)

        else:

            print('MS is already around, no need to copy it.')

​

    def setUp_fourants(self):

        # Input MS contain 2 scans and 16 spws. It has all data columns

        self.prefix = 'Four_ants_3C286'

        self.msfile = self.prefix + '.ms'  

        # Output files                      

        self.mmsfile = self.prefix + '.mms'

        fpath = os.path.join(datapath,self.msfile)

#         if os.path.lexists(fpath):        

#             shutil.copytree(fpath, self.msfile, symlinks=True)

#         else:

#             self.fail('Data does not exist -> '+fpath)

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

            shutil.copytree(fpath, self.msfile, symlinks=True)

        else:

            print('MS is already around, no need to copy it.')

​

    def setUp_floatcol(self):

        # 15 rows, 3 scans, 9 spw, mixed chans, XX,YY, FLOAT_DATA col

        self.prefix = "SDFloatColumn"

        self.msfile = 'SDFloatColumn.ms'

        # Output files