casa6

from __future__ import absolute_import

from __future__ import print_function

import os

import shutil

import unittest

import numpy

​

from casatasks.private.casa_transition import is_CASA6

if is_CASA6:

    from casatools import ctsys, image

    from casatasks import makemask

    from casatasks.private.imtools import pixelmask2cleanmask   ##### <----<<< this dependency should be removed

​

    ctsys_resolve = ctsys.resolve

​

    _ia = image( )

else:

    from recipes.pixelmask2cleanmask import *

    from tasks import *

    from taskinit import *

    from __main__ import default

​

    dataRoot = os.path.join(os.environ.get('CASAPATH').split()[0],'casatestdata/')

    def ctsys_resolve(apath):

        return os.path.join(dataRoot,apath)

​

    _ia = iatool( )

​

"""

Unit tests for task makemask 

​

"""

​

datapath = ctsys_resolve('unittest/makemask/')

​

#debug=True

debug=False

​

class makemaskTestBase(unittest.TestCase):

    """ base class for makemask unit test"""

    def compareimpix(self,refimage,inimage,evalmask=False):

        """

        do pixel by pixel comparison of the cube image

        returns true if identical in terms of pixel values

        and shape (does not check coordinates)

        """

        _ia.open(refimage)

        refdata=_ia.getchunk(getmask=evalmask)

        refshp=list(_ia.shape())

        _ia.close()

        _ia.open(inimage)

        indata=_ia.getchunk(getmask=evalmask)

        inshp=list(_ia.shape())

        _ia.close()

        if inshp==refshp:

            for i in range(inshp[3]):

                for j in range(inshp[2]):

                    for k in range(inshp[1]):

                        for l in range(inshp[0]):

                            if not indata[l][k][j][i]==refdata[l][k][j][i]:

                                return False

        return True

​