casa6

​

# Trivial tests of asdmsummary.

# Tests that asdmsummary works without dying on a few ASDMs from 

# different telescopes.  Also checks that the number of new log rows written

# by asdmsummary matches the expected number for that ASDM.

​

from __future__ import absolute_import

import os

​

from casatasks.private.casa_transition import is_CASA6

if is_CASA6:

    from casatasks import asdmsummary, casalog

else:

    from __main__ import default

    from tasks import asdmsummary

    from taskinit import casalog

​

import unittest

​

def logfileLen():

    # count the lines in the current log file

    result = 0

    logfile = casalog.logfile()

    if os.path.isfile(logfile):

        with open(logfile) as f:

            for result, l in enumerate(f,1):

                pass

    return result

​

class asdmsummary_test(unittest.TestCase):

​

    # trivial tests that just demonstrate it doesn't fail completely

    # also now counts the number of lines written to the log file against expected count

​

    # CASA5 needs to know where the data is

    # CASA5 spits out 8 additional lines to the log that CASA6 does not produce

    dataPath = ""

    extraLines = 0

    if not is_CASA6:

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

        extraLines = 8

​

    def doASDMSummary(self, asdmpath, expectedLogLines):

        # run asdmsummary, expepctedLogLines is the expected number of new log lines

        logLength = logfileLen()

        asdmsummary(os.path.join(self.dataPath,asdmpath))

        newLines = logfileLen()-logLength

        self.assertEqual(newLines,expectedLogLines+self.extraLines)

​

    def setUp(self):

        if is_CASA6:

            pass

        else:

            default(asdmsummary)

​

    def tearDown(self):

        pass

​

    def test_alma_asdm(self):

        ''' ALMA M51 data'''

        # used in test_importasdm, test_importasdm_mms.

        self.doASDMSummary('uid___X5f_X18951_X1',166)