casa6

import os

import sys

import shutil

import commands

import numpy

from __main__ import default

from tasks import *

from taskinit import *

from cleanhelper import *

import unittest

​

'''

Unit tests for cleanhelper class members.

'''

class cleanhelper_test(unittest.TestCase):

​

    # Input and output names

    # use simulated data with 3 point sources

    msfile = 'simptsrcs.ms'

    outlierfilename='outlier.txt'

    newoutlierfilename='newoutlier_v0.txt'

    newoutlierfile2name='newoutlier_manyfields.txt'

    newoutlierreffilename='newoutlier_ref.txt'

    boxfilename='cleanhelpertest-sf.box'

    boxfilename2='boxf.txt'

    rgnfilename='box0.rgn'

    rgntextfilename='regmask.txt'

    imgname='cleanhelpertest'

    # some other fixed parameters

    imsize=[300,300]

    cell='0.1arcsec'

    stokes='I'

    mode='mfs'

    spw=''

    nchan=1

    start=0

    width=1

    restfreq=''

    field=''

    phasecenter="J2000 17:45:40.0 -29.00.28"

    refpath = os.environ.get('CASAPATH').split()[0] + '/casatestdata/unittest/cleanhelper/cleanhelper_reference/'

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

    outlierfile=datapath+outlierfilename 

    newoutlierfile=datapath+newoutlierfilename 

    newoutlierfile2=datapath+newoutlierfile2name 

    newoutlierreffile=refpath+newoutlierreffilename 

    boxfile=datapath+boxfilename

    boxfile2=datapath+boxfilename2

    rgnfile=datapath+rgnfilename

    rgntextfile=datapath+rgntextfilename

    res = None

​

    def setUp(self):

        usescratch=False

        mosweight=False

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

            os.system('rm -rf ' + self.msfile)

        shutil.copytree(self.datapath+self.msfile, self.msfile)

​

        self.imset = cleanhelper(im, self.msfile, (usescratch or mosweight))

        self.imset.datsel(field=self.field, spw=self.spw, nchan=self.nchan,

                         start=self.start, width=self.width,

                         timerange='', uvrange='', antenna='', scan='', 

                         observation='',intent='', usescratch=usescratch)

​

​

    def tearDown(self):

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

            os.system('rm -rf ' + self.msfile)

​

        im.close()

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

    def getpixval(self,img,pixel):

        ia.open(img)

        px = ia.pixelvalue(pixel)

        ia.close()

        return px['value']['value']

​

    def comparemask(self,maskimage,refmask):

        """

        compare the input image with reference image

        return true if pix values are identical

​

        """

        ia.open(maskimage)

        maskvals = ia.getchunk()

        ia.close()

        ia.open(refmask)

        refvals= ia.getchunk()