casa6

########################################################################

# test_task_vishead.py

#

# Copyright (C) 2018

# Associated Universities, Inc. Washington DC, USA

#

# This script is free software; you can redistribute it and/or modify it

# under the terms of the GNU Library General Public License as published by

# the Free Software Foundation; either version 2 of the License, or (at your

# option) any later version.

#

# This library is distributed in the hope that it will be useful, but WITHOUT

# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Library General Public

# License for more details.

#

# [Add the link to the JIRA ticket here once it exists]

#

# Based on the requirements listed in plone found here:

# https://casadocs.readthedocs.io/en/stable/api/tt/casatasks.information.vishead.html

#

#

##########################################################################

import sys

import os

import unittest

import shutil

import numpy

​

import casatools

from casatasks import vishead, casalog

​

def copyDirectory(src, dest):

    try:

        shutil.copytree(src, dest)

    # Directories are the same

    except shutil.Error as e:

        print('Directory not copied. Error: %s' % e)

    # Any error saying that the directory doesn't exist

    except OSError as e:

        print('Directory not copied. Error: %s' % e)

​

def chmod_recursive(path, mode):

​

    os.chmod(path, mode)

    for root, dirs, files in os.walk(path):

        for d in dirs:

            os.chmod(os.path.join(root, d), mode)

        for f in files:

            os.chmod(os.path.join(root, f), mode)

​

datapath = casatools.ctsys.resolve('unittest/vishead/')

casaimagepath = casatools.ctsys.resolve('unittest/vishead/ngc5921.ms')

​

# used in a type comparison

numpy_str_ = numpy.str_

​

logpath = casalog.logfile()

​

# Pick up alternative data directory to run tests on MMSs

testmms = False

if 'TEST_DATADIR' in os.environ:

    DATADIR = str(os.environ.get('TEST_DATADIR'))+'/vishead/'