casa6

from __future__ import absolute_import

from __future__ import print_function

​

import unittest

import os

import math

import sys

​

from casatasks.private.casa_transition import is_CASA6

if is_CASA6:

    from casatasks import sdpolaverage

    from casatasks.private.sdutil import tbmanager

    from casatools import ctsys

    datapath = ctsys.resolve('unittest/sdpolaverage/')

​

    # default isn't used in casatasks

    def default(atask):

        pass

​

else:

    from tasks import sdpolaverage

    from __main__ import default

    from sdutil import tbmanager

​

    # Define the root for the data files

    datapath = os.environ.get('CASAPATH').split()[0] + "/casatestdata/unittest/sdpolaverage/"

​

​

def weighToSigma(weight):

    if weight > sys.float_info.min:

        return 1.0 / math.sqrt(weight)

    else:

        return -1.0

​

​

def sigmaToWeight(sigma):

    if sigma > sys.float_info.min:

        return 1.0 / math.pow(sigma, 2)

    else:

        return 0.0

​

​

def check_eq(val, expval, tol=None):

    """Checks that val matches expval within tol."""

#    print val

    if type(val) == dict:

        for k in val:

            check_eq(val[k], expval[k], tol)

    else:

        try:

            if tol and hasattr(val, '__rsub__'):

                are_eq = abs(val - expval) < tol

            else:

                are_eq = val == expval

            if hasattr(are_eq, 'all'):

                are_eq = are_eq.all()

            if not are_eq:

                raise ValueError('!=')

        except ValueError:

            errmsg = "%r != %r" % (val, expval)

            if (len(errmsg) > 66):  # 66 = 78 - len('ValueError: ')

                errmsg = "\n%r\n!=\n%r" % (val, expval)

            raise ValueError(errmsg)