casa6

###########################################################################

# test_task_plotweather.py

# Copyright (C) 2018

# Associated Universities, Inc. Washington DC, USA.

#

# This script is free software; you can redistribute it and/or modify it

# under the terms of the GNU Library General Public License as published by

# the Free Software Foundation; either version 2 of the License, or (at your

# option) any later version.

#

# This library is distributed in the hope that it will be useful, but WITHOUT

# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Library General Public

# License for more details.

#

#

# Based on the requirements listed in casadocs found here:

# https://casadocs.readthedocs.io/en/stable/api/tt/casatasks.visualization.plotweather.html

#

##########################################################################

import os

import string

import sys

import shutil

import unittest

​

from casatasks import plotweather

from casatools import ctsys

​

'''

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 (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'''

        with self.assertRaises(AssertionError):

            plotweather()

​

    def test1(self):

        '''Test 1: Bad input file'''

        badmsfile = 'badfile.ms'

        with self.assertRaises(AssertionError):

            plotweather(vis=badmsfile)

​

#    @unittest.skipIf("failure, 0.005426051322080905 != 0.0054234724819465846 within 7 places")

    def test2(self):

        '''Test 2: ms with no weather, no plot '''

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

            shutil.rmtree(self.msNoWeatherfile)

        shutil.copytree(ctsys.resolve(os.path.join("unittest/plotweather",self.msNoWeatherfile)), self.msNoWeatherfile)

​

        opac = plotweather(vis=self.msNoWeatherfile, plotName=self.fig)

        self.assertIsNotNone(opac)

        self.assertAlmostEqual(opac[0], 0.0054260513220809048)

        self.assertFalse(os.path.exists(self.fig))

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

            shutil.rmtree(self.msNoWeatherfile)

​

#    @unittest.skipIf("failure, 1.3931958371884026 != 1.3867727940788754 within 7 places")

    def test3(self):

        '''Test 3: Good input file and output exists'''

        res = plotweather(vis=self.msfile, plotName=self.fig)

        self.assertIsNotNone(res)