casa6

from __future__ import absolute_import

from __future__ import print_function

import os

import sys

import shutil

import unittest

​

from casatasks.private.casa_transition import is_CASA6

if is_CASA6:

    from casatools import ctsys, table, quanta

    from casatasks import fixvis, imstat, tclean, split

​

    _tb = table()

    _qa = quanta()

​

    ctsys_resolve = ctsys.resolve

else:

    from __main__ import default

    from tasks import *

    from taskinit import *

​

    # not local tools

    _tb = tb

    _qa = qa

​

    def ctsys_resolve(apath):

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

        return os.path.join(dataPath,apath)

​

'''

Unit tests for task fixvis.

​

Features tested:                                                       

  1. Do converted directions in the FIELD table have the right shape? 

  2. Does the phase center shifting result in the expected shifts?

  3. Does the distances parameter work

​

Note: The equinox_vis regression is a more general test of fixvis.

'''

datapath = ctsys_resolve('unittest/fixvis/')

inpms = '0420+417.ms'

outms = 'output.ms'

inpms2 = 'twocenteredpointsources.ms'

outms2 = 'testx.ms'

​

class fixvis_test1(unittest.TestCase):

    def setUp(self):

        res = None

        if not os.path.exists(inpms):

            shutil.copytree(os.path.join(datapath,inpms),inpms)

        if not os.path.exists(inpms2):

            shutil.copytree(os.path.join(datapath,inpms2),inpms2)

        if not is_CASA6:

            default(fixvis)

        shutil.rmtree(outms, ignore_errors=True)

        shutil.rmtree(outms2, ignore_errors=True)

    def tearDown(self):

        shutil.rmtree(inpms)

        shutil.rmtree(inpms2)

        shutil.rmtree(outms, ignore_errors=True)

        shutil.rmtree(outms2, ignore_errors=True)

        os.system('rm -rf test[yz]*')

​

    def _fixvis_and_get_stats(self, phasecent, dist=""):

        refcode = 'J2000'

        shutil.rmtree(outms2, ignore_errors=True)

        mystats = ''

        try:

            fixvis(inpms2, outms2, field='0', refcode=refcode,

                   phasecenter=phasecent, distances=dist)

            mystats = self._get_stats(0, 'testy')

        except Exception as exc:

            self.fail("*** Unexpected error *** {}".format(exc))

​

        return mystats

​

    def _get_stats(self, fld, imgname):

        os.system('rm -rf ' + imgname + '*')

​

        tclean( vis=outms2, imagename=imgname, field=str(fld), threshold='0.1mJy',

                deconvolver='clark', gridder='standard', mask='user', imsize=[128,128],

                cell=['0.10000008arcsec','0.10000008arcsec'], weighting='natural',

                uvtaper=[] )

        return imstat(imgname + '.image')

​

    def test1(self):

        '''Test1: Do converted directions in the FIELD table have the right shape?'''

        refcode = 'J2000'

        self.res = fixvis( inpms, outms, refcode=refcode )

        _tb.open(outms + '/FIELD')