casa / casa6 / dc438581138

Commits

Takeshi Nakazato authored dc438581138 18 Nov 2020

CAS-13160 tweak antenna selection to include autocorr

casa5/gcwrap/python/scripts/tests/test_sdatmcor.py

Modified

-1
+import glob
-2
+import os
-3
+import numpy as np
-4
+import shutil
-5
+import sys
-6
+import unittest
-7
-8
+from casatasks.private.casa_transition import is_CASA6
-9
+if is_CASA6:
-10
+    from casatasks import sdatmcor
-11
+    import casatasks.private.task_sdatmcor as sdatmcor_impl
-+
+    import casatasks.private.sdutil as sdutil
-13
+    # default isn't used in casatasks
-14
-15
+    def default(atask):
-16
+        pass
-17
+    # for testhelper import
-18
+    sys.path.append(
-19
+        os.path.dirname(
-20
+            os.path.abspath(
-21
+                os.path.dirname(__file__))))
-22
+    from casatasks.private.sdutil import tbmanager
-31
-32
+else:
-33
+    from tasks import sdatmcor
-34
+    import task_sdatmcor as sdatmcor_impl
-35
-36
+    from tasks import gencal, applycal
-37
+    from __main__ import default
-38
+    from sdutil import tbmanager
-39
+    from taskinit import cbtool as calibrater
-40
+    from taskinit import mstool
-+
+    import sdutil
-42
-43
+    # Define the root for the data files
-44
+    datapath = os.environ.get('CASAPATH').split()[0] + ''
-45
-46
+    def ctsys_resolve(apath):
-47
+        dataPath = os.path.join(os.environ['CASAPATH'].split()[0], 'data')
-48
+        subdir_hints = ['', 'casa-data-req']
-49
+        for subdir in subdir_hints:
-50
+            path = os.path.join(dataPath, subdir, apath)
-51
+            if os.path.exists(path):
-448
+    def test_sdatmcor_intent_selection(self):
-449
+        '''test intent selection: test if selection of ON_SOURCE data (i.e. excluding OFF_SOURCE data) still works'''
-450
+        sdatmcor(infile=self.infile, outfile=self.outfile, intent='OBSERVE_TARGET#ON_SOURCE*', datacolumn='data')
-451
+        self.check_result({19: True, 23: True}, on_source_only=True)
-452
-453
+    def test_sdatmcor_spw_process_less_than_20_select_all(self):
-454
+        '''test data selection: specify invalid spw to process'''
-455
+        sdatmcor(infile=self.infile, spw='<20', outputspw='', outfile=self.outfile, datacolumn='data')
-456
+        self.check_result({19: True, 23: False})
-457
-+
+    def test_sdatmcor_antenna_selection(self):
-+
+        """Test antenna selection"""
-+
+        sdatmcor(infile=self.infile, antenna='PM02', outfile=self.outfile)
-+
+        self.check_result({19: True, 23: True})
-+
-+
+    def test_sdatmcor_msselect(self):
-+
+        """Test msselect"""
-+
+        sdatmcor(infile=self.infile, msselect='ANTNNA1 == 1', outfile=self.outfile)
-+
+        self.check_result({19: True, 23: True})
-+
-468
+    def test_sdatmcor_gainfactor_float(self):
-469
+        """test gainfactor: float input"""
-470
+        gainfactor = 10.0
-471
+        apply_gainfactor(self.infile, 19, gainfactor)
-472
+        apply_gainfactor(self.infile, 23, gainfactor)
-473
+        sdatmcor(infile=self.infile, outfile=self.outfile, datacolumn='data', gainfactor=gainfactor)
-474
+        self.check_result({19: True, 23: True})
-475
-476
+    def test_sdatmcor_gainfactor_dict(self):
-477
+        """test gainfactor: dict input"""
-513
+                    expected,
-514
+                    msg='failed: input {} result {} expected {}'.format(
-515
+                        spw, actual, expected
-516
+                    )
-517
+                )
-518
+            else:
-519
+                # error cases
-520
+                with self.assertRaises(expected):
-521
+                    actual = sdatmcor_impl.parse_spw(self.infile, spw)
-522
-+
+    def test_tweak_antenna_selection(self):
-+
+        """Test tweak of antenna selection"""
-+
+        # common test cases
-+
+        test_cases0 = [
-+
+            ('', ''),
-+
+            ('PM02', 'PM02&&&'),
-+
+            ('PM02&', 'PM02&&&'),
-+
+            ('PM02&&', 'PM02&&'),
-+
+            ('PM02&&&', 'PM02&&&'),
-+
+            ('0', '0&&&'),
-+
+            ('0&', '0&&&'),
-+
+            ('0&&', '0&&'),
-+
+            ('0&&&', '0&&&'),
-+
+            ('0&1', '0&&&;1&&&'),
-+
+            ('0&&1', '0&&&;1&&&'),
-+
+            ('0;1', '0&&&;1&&&'),
-+
+        ]
-+
+        # specific to convert_antenna_spec_autocorr
-+
+        test_cases1 = [
-+
+            ('0&;1&&', '0&&&;1&&'),
-+
+            ('0&&&;1', '0&&&;1&&&')
-+
+        ]
-+
+        for antenna, expected in test_cases0 + test_cases1:
-+
+            actual = sdutil.convert_antenna_spec_autocorr(antenna)
-+
+            self.assertEqual(actual, expected)
-+
+        # specific to get_antenna_selection_include_autocorr
-+
+        test_cases2 = [
-+
+            ('0&;1&&', '0&;1&&'),
-+
+            ('0&&&;1', '0&&&;1'),
-+
+        ]
-+
+        for antenna, expected in test_cases0 + test_cases2:
-+
+            actual = sdutil.get_antenna_selection_include_autocorr(self.infile, antenna)
-+
+            self.assertEqual(actual, expected)
-+
-557
-558
+def suite():
-559
+    return [test_sdatmcor]
-560
-561
-562
+if is_CASA6:
-563
+    if __name__ == '__main__':
-564
+        unittest.main()
-565
+"""
-566
+History

casa6

Commits

Jira Issues

Add shortcut