casa6

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

# test_task_predictcomp.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.

#

# Features tested here:

#   objname: error for unsupported object vs supported object

#          non-visible case

#   standard: wrong standard vs correct standard

#   minfreq/maxfreq: wrong unit vs correct unit

#   output: check for the cl file

#   antennalist: use of the configuration file to plot 'observed' visibility

#              amplitudes vs uvdist. GUI is turned off.

#

# Based on the requirements listed in casadocs found here:

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

#

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

import os

import shutil

import numpy as np

import unittest

​

from casatools import ctsys

from casatasks import predictcomp, casalog

​

datapath = ctsys.resolve('alma/simmos/')

​

class predictcomp_test(unittest.TestCase):

​

    def setUp(self):

        self.res=None

​

    def tearDown(self):

        #pass

        os.system('rm -rf *.cl')

​

    def test_default(self):

        '''predictcomp: test defaults'''

        with self.assertRaises(ValueError):

            predictcomp()

    def test_invalid_objname(self): 

        '''predictcomp: invalid objname'''

        with self.assertRaises(RuntimeError):

            predictcomp(objname='Moon', minfreq='100GHz',maxfreq='120GHz')

​

    def test_valid_objname(self):

        '''predictcomp: valid objname'''

        self.res=predictcomp(objname='Titan', epoch='2017/09/01/00:00', minfreq='100GHz',maxfreq='120GHz',

                             standard='Butler-JPL-Horizons 2012') 

        print("type(self.res) = ",type(self.res))

        self.assertTrue(type(self.res)==dict)