casa6

import os

import sys

import math

import shutil

import string

import time

import numpy as np

import math

​

try:

    # CASA 6

    from casatools import table

    # Most of this helper file is about table operations. The ms and image tools are used

    # only for a couple of functions (for which there might be a better place)

    from casatools import ms, image

​

    tb_local = table()

    tb_local2 = table()

    ms_local = ms()

    image_local = image()

except ImportError:

    # CASA 5

    from taskinit import tbtool

    from taskinit import mstool, iatool

​

    tb_local = tbtool()

    tb_local2 = tbtool()

    ms_local = mstool()

    image_local = iatool()

​

​

'''

A set of common helper functions for unit tests:

   compTables - compare two CASA tables

   compVarColTables - Compare a variable column of two tables

   DictDiffer - a class with methods to take a difference of two 

                Python dictionaries

   verify_ms - Function to verify spw and channels information in an MS   

   create_input - Save the string in a text file with the given name           

'''

​

def phasediffabsdeg(c1, c2):

    try:

        a = c1.imag

        a = c2.imag

    except:

        print("Phase difference of real numbers is always zero.")

        return 0.

​

    a = math.atan2(c1.imag, c1.real)

    b = math.atan2(c2.imag, c2.real)

    diff = abs(a-b)

    if diff>np.pi:

        diff = 2*np.py - diff

    return diff/np.pi*180. # (degrees)

​

def compTables(referencetab, testtab, excludecols, tolerance=0.001, mode="percentage", startrow = 0, nrow = -1, rowincr = 1):

​

    """

    compTables - compare two CASA tables

       referencetab - the table which is assumed to be correct

​