casa6

from __future__ import absolute_import

import unittest

import inspect

import re

import abc

​

# get is_python3 and is_CASA6

from casatasks.private.casa_transition import *

if is_CASA6:

    from casatasks import casalog

else:

    from taskinit import casalog

​

def skipUnlessHasParam(param):

    def wrapper(func):

        import functools

        @functools.wraps(func)

        def _wrapper(*args, **kwargs):

            task = args[0].task

            if is_python3:

                task_args = inspect.getargs(task.__call__.__code__).args

            else:

                task_args = inspect.getargs(task.__code__).args

            if isinstance(param, str):

                condition = param in task_args

                if is_python3:

                    reason = '%s doesn\'t have parameter \'%s\''%(str(task), param)

                else:

                    reason = '%s doesn\'t have parameter \'%s\''%(task.__name__, param)

            else:

                # should be a list

                condition = all([p in task_args for p in param])

                reason = '%s doesn\'t have parameter %s'%(task.__name__, map(lambda x: '\'' + str(x) + '\'', param))

            return unittest.skipUnless(condition, reason)(func)(*args, **kwargs)

        return _wrapper

    return wrapper

​

def skipIfNoChannelSelection(func):

    import functools

    @functools.wraps(func)

    def wrapper(*args, **kwargs):

        task = args[0].task

        condition = args[0].spw_channel_selection

        reason = '%s doens\'nt accept channel selection on parameter spw'%(task.__name__)

        return unittest.skipUnless(condition, reason)(func)(*args, **kwargs)

    return wrapper

​

class SelectionSyntaxTest(unittest.TestCase):

    """

    Base class for data selection syntax unit test.

​

    All the tests are defined in this class but all fail by

    default. Subclasses must override necessary tests.

    Unnecessary tests will be automatically skipped.

​

    In the test implementation, it is recommended to use

    run_task method instead of running task directly. It

    will check whether the test is properly implemented in

    terms of its intent.

    """

    __metaclass__ = abc.ABCMeta

    infile = None