test_sdlist.py

Go to the documentation of this file.

import os
import sys
import shutil
from __main__ import default
from tasks import *
from taskinit import *
import unittest
#
import listing

from sdlist import sdlist

class sdlist_test(unittest.TestCase):
    """
    Unit test for task sdlist.

    The list of tests:
    test00   --- default parameters (raises an error)
    test01   --- valid input filename and save output as a text file
    test02   --- test scanaverage=True
    test03   --- test overwrite=True
    """
    # Data path of input/output
    datapath=os.environ.get('CASAPATH').split()[0] + '/data/regression/unittest/sdlist/'
    # Input and output names
    infile1 = 'OrionS_rawACSmod_cal2123.asap'
    infile2 = 'OrionS_rawACSmod'
    infiles = [infile1, infile2]
    outroot = 'sdlist_test'
    refroot = "refsdlist"

    def setUp(self):
        for infile in self.infiles:
            if os.path.exists(infile):
                shutil.rmtree(infile)
            shutil.copytree(self.datapath+infile, infile)

        default(sdlist)


    def tearDown(self):
        for infile in self.infiles:
            if (os.path.exists(infile)):
                shutil.rmtree(infile)

    def _compareOutFile(self,out,reference):
        self.assertTrue(os.path.exists(out))
        self.assertTrue(os.path.exists(reference),
                        msg="Reference file doesn't exist: "+reference)
        self.assertTrue(listing.compare(out,reference),
                        'New and reference files are different. %s != %s. '
                        %(out,reference))


    def test00(self):
        """Test 0: Default parameters"""
        result=sdlist()
        self.assertFalse(result)

    def test01(self):
        """Test 1: Save output to an ascii file"""
        tid = "01"
        infile = self.infile1
        outfile = self.outroot+tid+".out"
        result = sdlist(infile=infile,outfile=outfile)
        self.assertEqual(result,None,
                         msg="The task returned '"+str(result)+"' instead of None")
        self._compareOutFile(outfile,self.datapath+self.refroot+tid+".txt")
        
    def test02(self):
        """Test 2: Test scanaverage=True"""
        tid = "02"
        infile = self.infile1
        outfile = self.outroot+tid+".out"
        result = sdlist(infile=infile,scanaverage=True,outfile=outfile)
        self.assertEqual(result,None,
                         msg="The task returned '"+str(result)+"' instead of None")
        self._compareOutFile(outfile,self.datapath+self.refroot+tid+".txt")

    def test03(self):
        """Test 3: Test overwrite=True"""
        tid = "03"
        infile = self.infile2
        outfile = self.outroot+tid+".out"
        result = sdlist(infile=infile,outfile=outfile)
        self.assertEqual(result,None,
                         msg="The task returned '"+str(result)+"' instead of None")
        self.assertTrue(os.path.exists(outfile),
                        msg="Output file doesn't exist after the 1st run.")
        # overwrite 'outfile'
        infile = self.infile1
        outfile = self.outroot+tid+".out"
        result = sdlist(infile=infile,outfile=outfile,overwrite=True)
        self.assertEqual(result,None,
                         msg="The task returned '"+str(result)+"' instead of None")
        self._compareOutFile(outfile,self.datapath+self.refroot+tid+".txt")
        
    def test04(self):
        """Test 4: Test overwrite=False"""
        tid = "04"
        infile = self.infile2
        outfile = self.outroot+tid+".out"
        result = sdlist(infile=infile,outfile=outfile)
        self.assertEqual(result,None,
                         msg="The task returned '"+str(result)+"' instead of None")
        self.assertTrue(os.path.exists(outfile),
                        msg="Output file doesn't exist after the 1st run.")
        # overwrite 'outfile'
        infile = self.infile1
        outfile = self.outroot+tid+".out"
        try:
            result = sdlist(infile=infile,outfile=outfile,overwrite=False)
            self.assertTrue(False,
                            msg='The task must throw exception')
        except Exception, e:
            pos=str(e).find('Output file \'%s\' exists.'%(outfile))
            self.assertNotEqual(pos,-1,
                                msg='Unexpected exception was thrown: %s'%(str(e)))
        
def suite():
    return [sdlist_test]