casa5

import os

import sys

import shutil

import re

import numpy

import math

import contextlib

​

from __main__ import default

from tasks import *

from taskinit import *

import unittest

#

import listing

import sdutil

​

from sdcal import sdcal

from partition import partition

​

try:

    from testutils import copytree_ignore_subversion

except:

    from tests.testutils import copytree_ignore_subversion

@contextlib.contextmanager

def mmshelper(vis, separationaxis='auto'):

    outputvis = vis.rstrip('/') + '.mms'

    os.system('rm -rf {0}*'.format(outputvis))

    try:

        partition(vis=vis, outputvis=outputvis, separationaxis=separationaxis)

        if os.path.exists(outputvis):

            yield outputvis

        else:

            yield None

    finally:

        os.system('rm -rf {0}*'.format(outputvis))

​

class sdcal_test(unittest.TestCase):

​

    """

    Unit test for task sdcal.

​

    The list of tests:

    test00  --- default parameters (raises an error)

    test01  --- spwmap comprising list

    test02  --- spwmap comprising dictionary

    test03  --- spwmap comprising others

    test04  --- there is no infile

    test05

    """

​

    # Data path of input

    datapath=os.environ.get('CASAPATH').split()[0]+ '/data/regression/unittest/tsdcal/'

​

    # Input 

    infile1 = 'uid___A002_X6218fb_X264.ms.sel'

    infiles = [infile1]

    tsystable = 'out.cal'

​

    def setUp(self):

        for infile in self.infiles:

            if os.path.exists(infile):

                shutil.rmtree(infile)

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

        default(sdcal)

​

    def tearDown(self):

        for infile in self.infiles:

            if (os.path.exists(infile)):

                shutil.rmtree(infile)

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

            shutil.rmtree(self.tsystable)

​

    def _compareOutFile(self,out,reference):

        self.assertTrue(os.path.exists(out))

        self.assertTrue(os.path.exists(reference),msg="Reference file doesn't exist: "+reference)

        self.assertTrue(listing.compare(out,reference),'New and reference files are different. %s != %s. '%(out,reference))

​

    def test00(self):

        """Test00:Check the identification of TSYS_SPECTRuM and FPARAM"""

​

        tid = "00"

        infile = self.infile1

        sdcal(infile=infile, calmode='tsys', outfile=self.tsystable)

        compfile1=infile+'/SYSCAL'

        compfile2=self.tsystable

​

        tb.open(compfile1)

        subt1=tb.query('', sortlist='ANTENNA_ID, TIME, SPECTRAL_WINDOW_ID', columns='TSYS_SPECTRUM')

        tsys1=subt1.getcol('TSYS_SPECTRUM')