casa6

'''

Unit tests for task split.

​

Features tested:

  1. Are the POLARIZATION, DATA_DESCRIPTION, and (to some extent) the

     SPECTRAL_WINDOW tables correct with and without correlation selection?

  2. Are the data shapes and values correct with and without correlation

     selection?

  3. Are the WEIGHT and SIGMA shapes and values correct with and without

     correlation selection?

  4. Is a SOURCE table with bogus entries properly handled?

  5. Is the STATE table properly handled?

  6. Are generic subtables copied over?

  7. Are CHAN_WIDTH and RESOLUTION properly handled in SPECTRAL_WINDOW when

     channels are being selected and/or averaged?

  8. The finer points of spw:chan selection.

​

Note: The time_then_chan_avg regression is a more "end-to-end" test of split.

'''

​

from __future__ import absolute_import

from __future__ import print_function

import os

import numpy

import re

import sys

import shutil

import filecmp

import unittest

​

from casatasks.private.casa_transition import *

if is_CASA6:

    ### for testhelper import

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

    from recipes.listshapes import listshapes

    from casatestutils import testhelper as th

    from casatasks import cvel, flagcmd, flagdata, importasdm, listobs, partition, split

    from casatools import ctsys, ms, msmetadata, table

    from casatasks.private.parallel.parallel_task_helper import ParallelTaskHelper

​

    ctsys_resolve = ctsys.resolve

​

    # default isn't part of CASA6

    def default(atask):

        pass

else:

    from __main__ import default

    from recipes.listshapes import listshapes

    from casatestutils import testhelper as th

    from tasks import cvel, flagcmd, flagdata, importasdm, listobs, partition, split

    from taskinit import mstool as ms

    from taskinit import msmdtool as msmetadata

    from taskinit import tbtool as table

    from parallel.parallel_task_helper import ParallelTaskHelper

    from casa_stack_manip import stack_frame_find

​

    def ctsys_resolve(apath):

        dataPath = os.path.join(os.environ['CASAPATH'].split()[0],'casatestdata/')

        return os.path.join(dataPath,apath)

​

# common function to get a dictionary item iterator

if is_python3:

    def lociteritems(adict):

        return adict.items()

else:

    def lociteritems(adict):

        return adict.iteritems()

​

datapath = ctsys_resolve('unittest/split/')

​

# Pick up alternative data directory to run tests on MMSs

testmms = False

if 'TEST_DATADIR' in os.environ:

    testmms = True

    DATADIR = str(os.environ.get('TEST_DATADIR'))

    if os.path.isdir(DATADIR):

        datapath = DATADIR+'/split/'

​

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

​

if 'BYPASS_PARALLEL_PROCESSING' in os.environ:

    ParallelTaskHelper.bypassParallelProcessing(1)

​

'''

    Start of old tests, which are the same as test_split.

    class SplitFlags on are new tests.

'''

​

# local copy of tool

tblocal = table()

​