casa6

from __future__ import absolute_import

import os

import string

import sys

import shutil

import unittest

​

from casatasks.private.casa_transition import is_CASA6

if is_CASA6:

    from casatasks import plotweather

    from casatools import ctsys

    ctsys_resolve = ctsys.resolve

else:

    from __main__ import default

    from tasks import *

    from taskinit import *

​

    def ctsys_resolve(apath):

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

        return os.path.join(dataPath,apath)

​

'''

Unit tests for task plotweather. It tests the following parameters:

    vis:                wrong and correct values

    seasonal_weight:    default (0.5) and other values

    doPlot:             default (True) and False

    plotName:           if output is created; test formats

​

    return value:       [opacity] (type='list')

'''

​

class plotweather_test(unittest.TestCase):

​

    # Input MS, must have WEATHER table

    msfile = 'nep2-shrunk.ms'

    msNoWeatherfile = 'ngc5921_ut.ms'

    # output plots

    fig = '/tmp/plotweathertest.png'

    defaultFig = msfile + ".plotweather.png"

​

    def setUp(self):

        if not is_CASA6:

            default(plotweather)

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

            shutil.rmtree(self.msfile)

        shutil.copytree(ctsys_resolve(os.path.join("unittest/plotweather",self.msfile)), self.msfile)

​

    def tearDown(self):

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

            shutil.rmtree(self.msfile)

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

            os.remove(self.fig)

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

            os.remove(self.defaultFig)

​

    def test0(self):

        '''Test 0: Default parameters'''

        if is_CASA6:

            with self.assertRaises(AssertionError):

                plotweather()

        else:

            with self.assertRaises(RuntimeError):

                opac = plotweather()