casa6

from __future__ import absolute_import

from __future__ import print_function

import os

import numpy

import sys

import shutil

import unittest

​

from casatasks.private.casa_transition import *

if is_CASA6:

    from casatools import ctsys, table

    from casatasks import uvcontsub

    from casatasks.private.parallel.parallel_task_helper import ParallelTaskHelper

​

    tb = table( )

​

    ctsys_resolve = ctsys.resolve

else:

    from __main__ import default

    from tasks import *

    from taskinit import *

    from parallel.parallel_task_helper import ParallelTaskHelper

​

    # uses the global tb tool

​

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

    def ctsys_resolve(apath):

        return os.path.join(dataRoot,apath)

​

'''

Unit tests for task uvcontsub.

​

Features tested:

  1. uvcontsub will use DATA if CORRECTED_DATA is absent.

  2. It gets the right answer for a known line + 0th order continuum.

  3. It gets the right answer for a known line + 4th order continuum.

  4. It gets the right answer for a known line + 0th order continuum,

     even when fitorder = 4.

'''

#uvcdatadir = 'uvcontsub' 

datapath = 'unittest/uvcontsub/'

​

# Pick up alternative data directory to run tests on MMSs

testmms = False

if 'TEST_DATADIR' in os.environ:   

    testmms = True

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

    if os.path.isdir(DATADIR):

        datapath = DATADIR

​

if 'BYPASS_PARALLEL_PROCESSING' in os.environ:

    ParallelTaskHelper.bypassParallelProcessing(1)

​

#Commented out for refactoring (eliminated test_split dependence)

#class UVContChecker(SplitChecker):

#    """

#    Base class for uvcontsub unit testing.

#    """

#    need_to_initialize = True

#    records = {}

#

#    def do_split(self, corrsel):

#        """

#        This is only called do_split because it comes from SplitChecker.

#        run_task (uvcontsub in this case) would have been a better name.

#        """

#        record = {}

#        try:

#            print "\nRunning uvcontsub"

#            uvran = uvcontsub(self.inpms, fitspw='0:0~5;18~23',

#                               fitorder=corrsel, want_cont=True

#                               )

#        except Exception, e:

#            print "Error running uvcontsub"

#            raise e

#        for spec in ('cont', 'contsub'):

#            specms = self.inpms + '.' + spec

#            tb.open(specms)

#            record[spec] = tb.getcell('DATA', 52)

#            tb.close()

#            shutil.rmtree(specms)

#        self.__class__.records[corrsel] = record

#        return uvran

​

class UVContsubUnitTestBase(unittest.TestCase):

    """

    uvcontsub unittest base class (refactored)

    """  

    def initialize(self,inpms):

        """

        initialize