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 task imcollapse 00034 # </summary> 00035 # 00036 # <reviewed reviwer="" date="" tests="" demos=""> 00037 # </reviewed 00038 # 00039 # <prerequisite> 00040 # <ul> 00041 # <li> <linkto class="task_imcollapse.py:description">imcollapse</linkto> 00042 # </ul> 00043 # </prerequisite> 00044 # 00045 # <etymology> 00046 # Test for the imcollapse task 00047 # </etymology> 00048 # 00049 # <synopsis> 00050 # Test the imcollapse task and the ia.collapse() method upon which it is built. 00051 # </synopsis> 00052 # 00053 # <example> 00054 # 00055 # This test runs as part of the CASA python unit test suite and can be run from 00056 # the command line via eg 00057 # 00058 # `echo $CASAPATH/bin/casapy | sed -e 's$ $/$'` --nologger --log2term -c `echo $CASAPATH | awk '{print $1}'`/code/xmlcasa/scripts/regressions/admin/runUnitTest.py test_imcollapse[test1,test2,...] 00059 # 00060 # </example> 00061 # 00062 # <motivation> 00063 # To provide a test standard for the imcollapse task to ensure 00064 # coding changes do not break the associated bits 00065 # </motivation> 00066 # 00067 00068 ########################################################################### 00069 import shutil 00070 import casac 00071 from tasks import * 00072 from taskinit import * 00073 from __main__ import * 00074 import unittest 00075 00076 00077 00078 00079 00080 class ia_convolve_test(unittest.TestCase): 00081 00082 def setUp(self): 00083 pass 00084 00085 def tearDown(self): 00086 pass 00087 00088 def test_stretch(self): 00089 """ ia.convolve(): Test stretch parameter""" 00090 yy = iatool() 00091 mymask = "maskim" 00092 yy.fromshape(mymask, [200, 200, 1, 1]) 00093 yy.addnoise() 00094 yy.done() 00095 kernel = "kernel" 00096 shape = [200,200,1,20] 00097 yy.fromshape(kernel, shape) 00098 yy.fromshape("", shape) 00099 yy.addnoise() 00100 yy.done() 00101 yy.fromshape("", shape) 00102 self.assertRaises( 00103 Exception, 00104 yy.convolve, "", mymask, mask=mymask + ">0", stretch=False 00105 ) 00106 zz = yy.convolve("", mymask, mask=mymask + ">0", stretch=True) 00107 self.assertTrue(type(zz) == type(yy)) 00108 yy.done() 00109 zz.done() 00110 00111 def suite(): 00112 return [ia_convolve_test]