casa6

import os

import sys

import shutil

from __main__ import default

from tasks import *

from taskinit import *

import unittest

import fnmatch

import sha

import time

​

# Paths for data

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

​

# Read the data sets from another directory

if os.environ.has_key('TEST_DATADIR'):  

    DATADIR = str(os.environ.get('TEST_DATADIR'))

    if os.path.isdir(DATADIR):

        datapath = DATADIR

    else:

        print 'WARN: directory '+DATADIR+' does not exist'

​

print 'plotcal tests will use data from '+datapath

​

class plotcal_test_base(unittest.TestCase):

​

#    Btabledir = '/bandpass/'

    Btable  = 'ngc5921.ref1a.bcal'

    Bms  = 'ngc5921.ms'

#    Gtabledir = '/gaincal/'

    Gtable  = "ngc5921.ref1a.gcal"

#    Ttabledir = "/wvrgcal/input/"

    Ttable  = "wvrgcalctest.W"

#    TSYStabledir = "/flagdata/"

    TSYStable  = "X7ef.tsys"

    fig = "/tmp/testplotcal.png"

​

    def setUp(self):

        self.removePlotfile()

        default(plotcal)

​

    def setUpBtable(self):

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

            shutil.rmtree(self.Btable)

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

​

    def setUpGtable(self):

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

            shutil.rmtree(self.Gtable)

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

​

    def setUpTtable(self):

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

            shutil.rmtree(self.Ttable)

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

​

    def setUpTSYStable(self):

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

            shutil.rmtree(self.TSYStable)

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

​

    def setUpBms(self):

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