casa5

import os

import sys

import shutil

from __main__ import default

from tasks import *

from taskinit import *

import testhelper as th

import unittest

​

'''

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

    vis:   wrong and correct values

    vis:   check that output column is created

    Other tests: check that theoretical and calculated values of column CORRECTED_DATA 

                 are the same.

'''

class oldhanningsmooth_test(unittest.TestCase):

    # Input and output names

    msfile = 'ngc5921_ut.ms'

    res = None

    out = 'oldhanningsmooth_test'

​

    def setUp(self):

        self.res = None

        default(oldhanningsmooth)

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

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

    def tearDown(self):

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

            os.system('rm -rf ' + self.msfile)

        if (os.path.exists('cvelngc.ms')):

            os.system('rm -rf cvelngc.ms')

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

            os.system('rm -rf ' + self.out)

        shutil.rmtree('mynewms.ms',ignore_errors=True)

​

    def test0(self):

        '''Test 0: Default values'''

        self.res = oldhanningsmooth()

        self.assertFalse(self.res)

    def test1(self):

        """Test 1: Wrong input MS should return False"""

        msfile = 'badmsfile'

        self.res = oldhanningsmooth(vis=msfile)

        self.assertFalse(self.res)

​

    def test2(self):

        '''Test 2: Check that output column is created'''

        self.res = oldhanningsmooth(vis=self.msfile, datacolumn='corrected')

        self.assertEqual(self.res,None)

        cd = th.getColDesc(self.msfile, 'CORRECTED_DATA')

        self.assertTrue(len(cd))

    def test3(self):

        '''Test 3: Theoretical and calculated values should be the same with datacolumn==CORRECTED'''

​

      # check correct flagging (just for one row as a sample)

        flag_col = th.getVarCol(self.msfile, 'FLAG')