casa5

import os

import sys

import shutil

from __main__ import default

from tasks import *

from taskinit import *

import unittest

import math

import numpy

import numbers

​

_rg = rgtool( )

​

#run using

# `which casa` --nologger --log2term -c `echo $CASAPATH | awk '{print $1}'`/code/xmlcasa/scripts/regressions/admin/runUnitTest.py --mem test_imstat

#

​

'''

Unit tests for task imstat.

'''

class imstat_test(unittest.TestCase):

    # Input and output names

    moment = 'moment_map.im'

    s150 = '150arcsec_pix.im'

    s15 = '15arcsec_pix.im'

    s0_015 = '0.015arcsec_pix.im'

    s0_0015 = '0.0015arcsec_pix.im'

    s0_00015 = '0.00015arcsec_pix.im'

    linear_coords = 'linearCoords.fits'

    fourdim = '4dim.im'

    kimage = "ktest.im"

    res = None

​

    def _compare(self, resold, resnew, helpstr):

        mytype = type(resold)

        self.assertTrue(mytype == type(resnew), helpstr + ": types differ")

        if mytype == dict:

            for k in resold.keys():

                self._compare(resold[k], resnew[k], helpstr)

        elif mytype == numpy.ndarray:

            oldarray = resold.ravel()

            newarray = resnew.ravel()

            self.assertTrue(

                len(oldarray) == len(newarray),

                helpstr + ": array lengths not equal"

            )

            for i in range(len(oldarray)):

                self._compare(oldarray[i], newarray[i], helpstr)

        elif mytype == str:

            self.assertTrue(

                resold == resnew,

                helpstr + ": string inequality, old = " + resold + ", new = " + resnew

            )

        elif isinstance(resold, numbers.Integral) or mytype == numpy.int32:

            self.assertTrue(

                resold == resnew,

                helpstr + ": integral inequality, old = " + str(resold) + ", new = " + str(resnew)

            )

        elif isinstance(resold, numbers.Real):

            self.assertTrue(

                resold == resnew

                or abs(resnew/resold - 1) < 1e-6,