casa6

from __future__ import absolute_import

from __future__ import print_function

import shutil

import unittest

import os

import filecmp

​

from casatasks.private.casa_transition import is_CASA6

if is_CASA6:

    from casatasks import flagcmd, flagdata, flagmanager

    from casatools import ctsys, agentflagger, table

    from collections import OrderedDict

​

    # Local copy of the agentflagger tool

    aflocal = agentflagger()

​

    # default isn't necessary in CASA6

    def default(atask):

        pass

​

    # Path for data

    datapath = ctsys.resolve("unittest/flagcmd/")

else:

    from tasks import flagcmd, flagdata, flagmanager

    from taskinit import aftool

    from taskinit import tbtool as table

    from __main__ import default

    from OrderedDictionary import OrderedDict

​

    # Local copy of the agentflagger tool

    aflocal = aftool()

​

    # Path for data

    datapath = os.environ.get('CASAPATH').split()[0] + "/casatestdata/unittest/flagcmd/"

​

#

# Test of flagcmd task. It uses flagdata to unflag and summary

#

​

def check_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,fname=''):

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

    if fname=='':

        inp = 'flagcmd.txt'

    else:

        inp=fname

​

    cmd = str_text

    # remove file first

    if os.path.exists(inp):

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

    # save it in a file