casa5

import os

import sys

import shutil

from __main__ import default

from tasks import *

from taskinit import *

import unittest

import fnmatch

import sha

import time

import matplotlib.pyplot as plt

import numpy as np

​

# Paths for data

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

altdatapath = os.environ.get('CASAPATH').split()[0] + "/data/regression/unittest/setjy/"

calpath = os.environ.get('CASAPATH').split()[0] + "/data/regression/unittest/gaincal/"

​

# Pick up alternative data directory to run tests on MMSs

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

    DATADIR = str(os.environ.get('TEST_DATADIR'))+'/plotms/'

    if os.path.isdir(DATADIR):

        datapath = DATADIR         

​

print 'plotms tests will use data from '+ datapath

​

class plotms_test_base(unittest.TestCase):

​

    testms  = "pm_ngc5921.ms"

    testms2 = "ngc5921.ms"

    testcaltable = 'ngc5921.ref1a.gcal'

    outputDir="/tmp/" + str(os.getpid()) + "/"

    plotfile_jpg = "/tmp/myplot.jpg"

    display = os.environ.get("DISPLAY")

    ms = os.path.join(outputDir, testms)

    ms2 = os.path.join(outputDir, testms2)

    caltable = os.path.join(outputDir, testcaltable)

​

    def cleanUp(self):

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

            shutil.rmtree(self.outputDir)

​

    def setUpdata(self):

        res = None

        default(plotms)

        if not os.path.exists(self.ms):

            shutil.copytree(os.path.join(datapath,self.testms), 

                    self.ms, symlinks=True)

​

    def tearDowndata(self):

        self.cleanUp()

        pm.setPlotMSFilename("")

​

    def setUpaltdata(self):

        if not os.path.exists(self.ms2):

            shutil.copytree(os.path.join(altdatapath,self.testms2),

                    self.ms2, symlinks=True)

​

    def setUpcaldata(self):

        res = None

        default(plotms)

        if not os.path.exists(self.ms2):

            shutil.copytree(os.path.join(calpath,self.testms2), 

                    self.ms2, symlinks=True)

        testcaltable = os.path.join(self.outputDir, self.caltable)

        if not os.path.exists(self.caltable):

            shutil.copytree(os.path.join(calpath, self.testcaltable),

                    self.caltable, symlinks=True)

​

    def checkPlotfile(self, plotfileName, minSize, maxSize=None):

        self.assertTrue(os.path.isfile(plotfileName), "Plot was not created")

        plotSize = os.path.getsize(plotfileName)

        print plotfileName, 'file size is', plotSize

        self.assertGreater(plotSize, minSize)

        if maxSize:

            self.assertLess(plotSize, maxSize)

    # plotms can return True if it catches an error, but makes no plot file

    def checkNoPlotfile(self, plotfileName):

        self.assertFalse(os.path.isfile(plotfileName), "Plot was created")

​

    def getFilecount(self, dirName, namePattern ):

        nameTarget = namePattern + '*'

        count = 0

        for  file in os.listdir( dirName ):

            if fnmatch.fnmatch( file, nameTarget):

                count = count + 1

        return count

    def removeFiles(self, dirName, namePattern ):

        nameTarget = namePattern + '*'