casa6

""" Class to wrap a test to use with unittest framework."""

​

import os

import commands

import sys

import shutil

import inspect

import re

import string

import traceback

import casac

import unittest

​

import tests.selection_syntax as selection_syntax

SyntaxTestClasses = [selection_syntax.SelectionSyntaxTest]

​

PYVER = str(sys.version_info[0]) + "." + str(sys.version_info[1])

​

AIPS_DIR = os.environ["CASAPATH"].split()[0]

DATA_DIR = AIPS_DIR+'/data'

​

SCRIPT_REPOS=AIPS_DIR + "/" + os.environ["CASAPATH"].split()[1] + '/lib/python' + PYVER + '/tests/'

​

if not os.access(SCRIPT_REPOS, os.F_OK):

    if os.access(AIPS_DIR+'/lib64', os.F_OK):

        SCRIPT_REPOS = AIPS_DIR+'/lib64/python' + PYVER + '/tests/'

    elif os.access(AIPS_DIR+'/lib', os.F_OK):

        SCRIPT_REPOS = AIPS_DIR+'/lib/python' + PYVER + '/tests/'

    else:            #Mac release

        SCRIPT_REPOS = AIPS_DIR+'/Resources/python/tests/'

​

class Helper():

    # This class is called when a test is not found. It will

    # raise an exception and make nose fail. This way, the not

    # found test will be counted as an error and won't be ignored.

    def __init__(self, name):

        self.tname = name

    def test_dummy(self):

        '''Helper function'''

        raise Exception, "Cannot find test %s"%self.tname

​

class UnitTest:

    def __init__(self,testname=''):

        """Take the name of a test file (without .py), wrap it and run"""

        self.testname = testname

        self.workdir = testname+'_work'

        self.scriptdir = SCRIPT_REPOS       

        self.datadir = [DATA_DIR]

        self.dataFiles = []

    def funcdesc(self):

        '''Name of test for FunctionTestCase'''

        return 'Test '+self.testname

    def funcSetup(self):        

        """Copy data files to local working directory"""

        dataFiles = self.dataFiles

        print 'Searching for input data in %s'%(self.datadir)

        for datafile in dataFiles: 

            file = self.locatedata(datafile, self.datadir)

            #if data already exist, remove them

            if(file != ''):

                os.system('rm -rf '+ self.workdir+'/'+datafile)

            if(os.path.isdir(file)):

                shutil.copytree(file, self.workdir+'/'+datafile)

            if(os.path.isfile(file)):

                shutil.copy(file, self.workdir+'/'+datafile)

    def funcTeardown(self):

        """Remove data files from working directory"""

        dataFiles = self.dataFiles

        for datafile in dataFiles:

            file = self.workdir+'/'+datafile

            os.system('rm -rf ' + file)

        self.dataFiles = []

​

    def getFuncTest(self):

        print '-------------- Unit Test for %s ---------------'%self.testname

        """Wrap a script using unittest"""

        testscript = self.searchscript(self.testname, self.scriptdir)

        # avoid creating a _work directory

        if (testscript == ""):

            return

​

        # copy test to local directory

        self.workdir = os.getcwd()