casa6

import os

import logging

import numpy

​

try:

    # CASA 6

    logging.debug("Importing CASAtools")

    import casatools

    tb = casatools.table()

    casa6 = True

​

except ImportError:

    # CASA 5

    logging.debug("Import casa6 errors. Trying CASA5...")

    from taskinit import tbtool

    tb = tbtool()

    casa5 = True

​

class Check(object):

    def __init__(self):

        self.val = False

        self.plotSize = ''

​

    def check_plotfile(self, plotfileName, min_size, max_size=None):

        '''

        Check if plotfile generated is cprrect size

            plotfileName --> Name of plotted Image

            min_size -- > Min Size of image

            max_size --> Max Size of image

            Return : True if image size > min_size ( and < max_size if max_size is provided )

        '''

​

        if os.path.isfile(plotfileName):

            self.plotSize = os.path.getsize(plotfileName) # Return the size, in bytes, of path.

            logging.info('{} file size is: {}'.format(plotfileName, self.plotSize))

            if self.plotSize > min_size:

                self.val = True

            if max_size is not None:

                if not self.plotSize < max_size:

                    self.val = False

        else:

            logging.critical("Plot was not created")

        return self.val

​

    def check_pixels(self, imagename='', loc=None, refval=None, rtol=1e-05, atol=1e-08):

        '''

            Check pixels in an image to a specified reference value

            @param imagename: input image file

            @param loc: The index of the image to compare to the refval

            @param refval: The reference value to compare the selected pixel(s) to

            @param rtol: The relative tolerance used in the numpy.isclose function

            @param atol: The absolute tolerance used in the numpy.isclose function

            @return: True if the shape and value of the refval and selected pixel match.

        '''

        if not isinstance(loc, str):

            raise TypeError('Please give target location in string list format "20,30,2:4"')

        if os.path.exists(imagename):

            tb.open(imagename)

            image = tb.getcol('map')

            tb.close()

            if not isinstance(refval, None):

                index = []

                to_slice = loc.split(',')