casa / casa6 / ebdc682afb9

Commits

David Mehringer authored ebdc682afb9 18 May 2022

add test to verify correct beam is place in metadata

casatasks/tests/tasks/test_task_imsmooth.py

Modified

-69
+        for j in range(shape[1]):
-70
+            y = shape[1]/2 - j
-71
+            xfac = x*x*fac/(xfwhm*xfwhm)
-72
+            yfac = y*y*fac/(yfwhm*yfwhm)
-73
+            values[i, j] = math.exp(-(xfac + yfac));
-74
+    return values
-75
-76
-77
+class imsmooth_test(unittest.TestCase):
-78
-+
+    cas_13827_in = 'cas_13827.im'
-+
+    cas_13827_out = 'cas_13827.conv'
-+
-82
+    def setUp(self):
-83
+        self.tst = 'imsmooth_'
-84
+        if(os.path.exists(image_names[0])):
-85
+            for file in image_names:
-86
+                os.system('rm -rf ' +file)
-87
-88
+        for file in image_names:
-89
+            os.system('cp -RH '+ os.path.join(datapath,file)+' ' + file)
-90
+        self.ia = image()
-91
+        for f in [targetres_im, tiny]:
-104
+        os.system('rm -rf sub'+'*')
-105
+        for ff in [targetres_im, tiny]:
-106
+            if(os.path.exists(ff)):
-107
+                os.system('rm -rf ' +ff)
-108
+        for ff in image_names:
-109
+            os.system('rm -rf ' + ff)
-110
+        if os.path.exists("test_image2dconvolver_multibeam.im"):
-111
+            shutil.rmtree("test_image2dconvolver_multibeam.im")
-112
+        if os.path.exists( 'garbage.rgn' ):
-113
+            os.remove('garbage.rgn')
-+
+        for im in (self.cas_13827_in, self.cas_13827_out):
-+
+            if os.path.exists(im):
-+
+                os.system(f'rm -rf {im}')
-+
-118
-119
+    ####################################################################
-120
+    # Incorrect inputs to parameters.  The parameters are:
-121
+    #    imagename
-122
+    #    outfile
-123
+    #    kernel
-124
+    #    major
-125
+    #    minor
-126
+    #    mask
-127
+    #    region
-1016
+                self.assertTrue(_near(gotbeam[j], expbeam[j], 2e-7))
-1017
+            self.assertTrue(abs(got.getchunk(blc=blc, trc=trc) - expec.getchunk()).max() < 3e-5)
-1018
+            got.done()
-1019
+            expec.done()
-1020
+        myia.done()
-1021
+        got.done()
-1022
-1023
+    def test_targetres(self):
-1024
+        """Test targetres parameter"""
-1025
+        myia = self.ia
--
+        imagename = self.tst+"tres1.im"
-+
+        imagename = f'{self.tst}tres1.im'
-1027
+        myia.fromshape(imagename, [100, 100])
-1028
+        csys = myia.coordsys()
-1029
+        csys.setunits(["arcsec", "arcsec"])
-1030
+        csys.setincrement([-1, 1])
-1031
+        myia.setcoordsys(csys.torecord())
-1032
+        myia.setrestoringbeam(major="6arcsec", minor="3arcsec", pa="0deg")
-1033
+        shape = myia.shape()
-1034
+        values = make_gauss2d(shape, 3.0, 6.0)
-1035
+        expected = make_gauss2d(shape, 5.0, 10.0)
-1036
+        myia.putchunk(values)
-1118
+                    subim.done()
-1119
+                    expected[:, :, k] = convim.getchunk()
-1120
+                    gotbeam = convim.restoringbeam()
-1121
-1122
+                    if targetres:
-1123
+                        self._compare_beams(gotbeam, {"major": major, "minor": minor, "pa": pa})
-1124
-1125
+                    ebeam.append(gotbeam)
-1126
+                    convim.done()
-1127
+                if targetres:
--
+                    outfile = self.tst+"tres3" + unit[0]
-+
+                    outfile = f'{self.tst}tres3{unit[0]}'
-1129
+                else:
--
+                    outfile = self.tst+"tres4" + unit[0]
-+
+                    outfile = f'{self.tst}tres4{unit[0]}'
-1131
+                run_imsmooth(
-1132
+                     imagename=imagename, kernel="gaussian",
-1133
+                     major=major, minor=minor, pa=pa, targetres=targetres,
-1134
+                     outfile=outfile
-1135
+                )
-1136
+                outia.open(outfile)
-1137
+                for k in range(shape[2]):
-1138
+                    gotbeam = outia.restoringbeam(channel=k)
-1139
+                    self._compare_beams(gotbeam, ebeam[k])
-1140
+                    maxdiff = (abs(outia.getchunk()-expected)).max()
-1141
+                    self.assertTrue(maxdiff < 1e-6)
-1142
+        myia.done()
-1143
+        outia.done()
-1144
-1145
+        myia.open(imagename)
-1146
+        myia.setrestoringbeam(
-1147
+            major="6arcsec", minor="3arcsec", pa="0deg"
-1148
+        )
-1149
+        myia.done()
--
+        outfile = self.tst+"tres6"
-+
+        outfile = f'{self.tst}tres6'
-1151
+        with self.assertRaises(RuntimeError):
-1152
+            run_imsmooth(
-1153
+                imagename=imagename, kernel="gaussian",
-1154
+                major="5.99arcsec", minor="2.99arcsec", pa="0deg",
-1155
+                targetres=True, outfile=outfile
-1156
+            )
-+
+        # CAS-13827, verify beam placed in output image header is the specified
-+
+        # beam, not the fitted beam
-+
+        myia.fromshape(self.cas_13827_in, [512, 512, 1, 1], overwrite=True)
-+
+        csys = myia.coordsys()
-+
+        inc = csys.increment()['numeric']
-+
+        inc[:2] /= 60
-+
+        csys.setincrement(inc)
-+
+        myia.setcoordsys(csys.torecord())
-+
+        csys.done()
-+
+        myia.setbrightnessunit('Jy/beam')
-+
+        myia.setrestoringbeam(major='3arcsec', minor='3arcsec', pa='0deg')
-+
+        myia.done()
-+
+        major = '3.2arcsec'
-+
+        minor = '3.1arcsec'
-+
+        pa = '60deg'
-+
+        imsmooth(
-+
+            self.cas_13827_in, outfile=self.cas_13827_out, targetres=True,
-+
+            major=major, minor=minor, pa=pa, overwrite=True
-+
+        )
-+
+        myia.open(self.cas_13827_out)
-+
+        beam = myia.restoringbeam()
-+
+        myia.done()
-+
+        self.assertEqual(
-+
+            _qa.quantity(beam['major']), _qa.quantity(major),
-+
+            'Incorrect beam in image metadata'
-+
+        )
-+
+        self.assertEqual(
-+
+            _qa.quantity(beam['minor']), _qa.quantity(minor),
-+
+            'Incorrect beam in image metadata'
-+
+        )
-+
+        self.assertEqual(
-+
+            _qa.quantity(beam['positionangle']), _qa.quantity(pa),
-+
+            'Incorrect beam in image metadata'
-+
+        )
-+
-1192
-1193
+    def test_overwrite(self):
-1194
+        """ test overwrite parameter """
-1195
+        myia = self.ia
-1196
+        outfile = self.tst+"test_overwrite.im"
-1197
+        myia.fromshape(outfile, [200, 200])
-1198
+        imagename = self.tst+"input_overwrite"
-1199
+        myia.fromshape(imagename, [200, 200])
-1200
+        myia.done()
-1201
+        run_imsmooth(

casa6

Commits

Jira Issues

Add shortcut