casa6

from __future__ import absolute_import

from __future__ import print_function

import os

import shutil

import unittest

import filecmp

import pprint

import numpy as np

from numpy import array

import ast

​

from casatasks.private.casa_transition import is_CASA6

if is_CASA6:

    import sys

​

    from casatasks import flagcmd, flagdata, mstransform, setjy, delmod, split

    from casatools import ctsys, agentflagger, table, measures, quanta

    from casatasks.private.parallel.parallel_task_helper import ParallelTaskHelper

    from casatasks.private import flaghelper as fh

​

    ### for testhelper import

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

    #import testhelper as th

​

    # CASA6 doesn't need defaul

    def default(atask):

        pass

​

    ctsys_resolve = ctsys.resolve

else:

    from tasks import flagcmd, flagdata, mstransform, setjy, delmod, split

    from taskinit import aftool as agentflagger

    from taskinit import tbtool as table

    from taskinit import metool as measures

    from taskinit import qatool as quanta

    from __main__ import default

    from parallel.parallel_task_helper import ParallelTaskHelper

    import flaghelper as fh

    #import testhelper as th

​

    def ctsys_resolve(apath):

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

        return os.path.join(dataPath,apath)

​

from casatestutils import testhelper as th

​

#from IPython.kernel.core.display_formatter import PPrintDisplayFormatter

​

#

# Test of flagdata modes

#

​

def func_test_eq(result, total, flagged):

​

    print("%s of %s data was flagged, expected %s of %s" % \

    (result['flagged'], result['total'], flagged, total))

    assert result['total'] == total, \

               "%s data in total; %s expected" % (result['total'], total)

    assert result['flagged'] == flagged, \

           "%s flags set; %s expected" % (result['flagged'], flagged)

​

def create_input(str_text, filename):

    '''Save the string in a text file'''

    inp = filename

    cmd = str_text

    # remove file first

    if os.path.exists(inp):

        os.system('rm -f '+ inp)

    # save to a file    

    with open(inp, 'w') as f:

        f.write(cmd)

    f.close()

    return

​

# Path for data

datapath = ctsys_resolve("unittest/flagdata/")

​

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

    if os.path.isdir(DATADIR):

        testmms = True

        datapath = DATADIR

​

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