casa6

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

# test_casatasks.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 in:

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

#

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

import os

import unittest

import numpy as np

from casatasks import importasdm, importfits, listobs, flagdata, gencal, setjy, fluxscale

from casatasks import gaincal, bandpass, mstransform, tclean, immoments, sdcal

from casatestutils import sparse_check

​

​

class BaseClass(unittest.TestCase):

    """ Base class with helper functions """

    def getdata(testfiles=None):

        # Download data for the tests

        sparse_check.download_data(testfiles)

​

​

class CasaTasksTests(BaseClass):

    """ Unit tests for CASA tasks

        The test script will download test data automatically to

        the local directory and will remove them afterwards

        """

​

    @classmethod

    def setUpClass(cls) -> None:

        cls.asdm = 'AutocorrASDM'

        cls.fitsimage = 'two_gaussian_model.fits'

        cls.listobs_ms = 'uid___X02_X3d737_X1_01_small.ms'

        cls.flagdata_ms = 'ngc5921.ms'

        cls.gencal_ms = 'tdem0003gencal.ms'

        cls.setjy_ms = 'ngc5921.ms'

        cls.fluxscale_ms = 'CalMSwithModel.ms'

        cls.fluxscale_gtable = 'ModelGcal.G0'

        cls.gaincal_ms = 'gaincaltest2.ms'

        cls.split_ms = 'Four_ants_3C286.ms'

        cls.tclean_ms = 'refim_oneshiftpoint.mosaic.ms'

        cls.immoments_img = 'n1333_both.image'

        cls.sdcal_ms = "otf_ephem.ms"

        cls.input_files = [

            cls.asdm, cls.fitsimage, cls.listobs_ms, cls.flagdata_ms,

            cls.gencal_ms, cls.fluxscale_ms, cls.fluxscale_gtable,

            cls.gaincal_ms, cls.split_ms, cls.tclean_ms, cls.immoments_img,

            cls.sdcal_ms

        ]

        # Fetch input data