00001 ########################################################################## 00002 # imfit_test.py 00003 # 00004 # Copyright (C) 2008, 2009 00005 # Associated Universities, Inc. Washington DC, USA. 00006 # 00007 # This script is free software; you can redistribute it and/or modify it 00008 # under the terms of the GNU Library General Public License as published by 00009 # the Free Software Foundation; either version 2 of the License, or (at your 00010 # option) any later version. 00011 # 00012 # This library is distributed in the hope that it will be useful, but WITHOUT 00013 # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 00014 # FITNESS FOR A PARTICULAR PURPOSE. See the GNU Library General Public 00015 # License for more details. 00016 # 00017 # You should have received a copy of the GNU Library General Public License 00018 # along with this library; if not, write to the Free Software Foundation, 00019 # Inc., 675 Massachusetts Ave, Cambridge, MA 02139, USA. 00020 # 00021 # Correspondence concerning AIPS++ should be adressed as follows: 00022 # Internet email: aips2-request@nrao.edu. 00023 # Postal address: AIPS++ Project Office 00024 # National Radio Astronomy Observatory 00025 # 520 Edgemont Road 00026 # Charlottesville, VA 22903-2475 USA 00027 # 00028 # <author> 00029 # Dave Mehringer 00030 # </author> 00031 # 00032 # <summary> 00033 # Test suite for the CASA tool method ia.rotate() 00034 # </summary> 00035 # 00036 # <reviewed reviwer="" date="" tests="" demos=""> 00037 # </reviewed 00038 # 00039 # <prerequisite> 00040 # <ul> 00041 # </ul> 00042 # </prerequisite> 00043 # 00044 # <etymology> 00045 # Test for the ia.rotate() tool method 00046 # </etymology> 00047 # 00048 # <synopsis> 00049 # Test the ia.rotate() tool method 00050 # </synopsis> 00051 # 00052 # <example> 00053 # 00054 # This test runs as part of the CASA python unit test suite and can be run from 00055 # the command line via eg 00056 # 00057 # `echo $CASAPATH/bin/casapy | sed -e 's$ $/$'` --nologger --log2term -c `echo $CASAPATH | awk '{print $1}'`/code/xmlcasa/scripts/regressions/admin/runUnitTest.py test_ia_rotate[test1,test2,...] 00058 # 00059 # </example> 00060 # 00061 # <motivation> 00062 # To provide a test standard for the ia.rotate() tool method to ensure 00063 # coding changes do not break the associated bits 00064 # </motivation> 00065 # 00066 00067 ########################################################################### 00068 import shutil 00069 import casac 00070 from tasks import * 00071 from taskinit import * 00072 from __main__ import * 00073 import unittest 00074 00075 class ia_rotate_test(unittest.TestCase): 00076 00077 def setUp(self): 00078 pass 00079 00080 def tearDown(self): 00081 pass 00082 00083 def test_stretch(self): 00084 """ ia.rotate(): Test stretch parameter""" 00085 yy = iatool() 00086 mymask = "maskim" 00087 yy.fromshape(mymask, [200, 200, 1, 1]) 00088 yy.addnoise() 00089 yy.done() 00090 shape = [200,200,1,20] 00091 yy.fromshape("", shape) 00092 yy.addnoise() 00093 self.assertRaises( 00094 Exception, 00095 yy.rotate, 00096 mask=mymask + ">0", stretch=False 00097 ) 00098 zz = yy.rotate( 00099 mask=mymask + ">0", stretch=True 00100 ) 00101 self.assertTrue(zz and type(zz) == type(yy)) 00102 yy.done() 00103 zz.done() 00104 00105 def suite(): 00106 return [ia_rotate_test]