casa6

###Base class for casa test scripts

###More complicated tests may inherit from this

import os

import string

import commands

import shutil

import sys

from  casac import *

import inspect

import re

import pdb

#from tasks import *  # execfile

#from taskinit import casalog

​

ia=casac.image()

ms=casac.ms()

tb=casac.table()

​

class testbase :

    def __init__(self, workdir=None):

        if workdir == None:

            workdir=os.getcwd()

        self.dataBaseDirectory=[workdir+'/Data']

        self.resultDirectory=workdir+'/Results'

        self.workingDirectory=workdir

        self.scriptRepository=workdir+'/Scripts'

        self.testsToRun=[]

        self.testList={}

        self.notest=False

​

    def createDirs(self):

        if os.access(self.resultDirectory, os.F_OK) is False:

            print self.resultDirectory+' does not exist, creating it'

            os.makedirs(self.resultDirectory)

        if os.access(self.workingDirectory, os.F_OK) is False:

            print self.workingDirectory+' does not exist, creating it'

            os.makedirs(self.workingDirectory)

    def setDataBaseDir(self, dir=['./Data']):

        if type(dir) != type(["directory", "list"]):

            raise TypeError, type(dir)

        self.dataBaseDirectory=dir

​

    def setResultDir(self, dir='./Results'):

        self.resultDirectory=dir

​

    def setWorkingDir(self, dir='./Temporaire'):

        self.workingDirectory=dir

    def setScriptsDir(self, dir='./Scripts'):

        self.scriptRepository=dir

​

    def cleanup(self):

        if os.path.isdir(self.workingDirectory):

            #print 'Removing '+ self.workingDirectory

            shutil.rmtree(self.workingDirectory)

​

    def locateTests(self, tests=None):

        theF=[]

        theFiles=[]

        if tests is None :

            # This does not work, need to search also for DIR/name.py

            # Also, searching in .../tests/<name>/regression.py is