casa / casa6 / 1fa18c65950

Commits

Takeshi Nakazato authored 1fa18c65950 09 May 2022

CAS-13713 add test for OMP_NUM_THREADS env var.

casatasks/tests/tasks/test_task_sdatmcor.py

Modified

-16
+#
-17
+# [Add the link to the JIRA ticket here once it exists]
-18
+#
-19
+# Based on the requirements listed in plone found here:
-20
+# https://casadocs.readthedocs.io/en/stable/api/tt/casatasks.single.sdatmcor.html
-21
+#
-22
+#
-23
+##########################################################################
-24
+import itertools
-25
+import os
-+
+import re
-27
+import shutil
-28
+import unittest
-29
-30
+import numpy as np
-31
-32
+from casatasks import applycal, casalog, gencal, sdatmcor
-33
+from casatasks.private.sdutil import (convert_antenna_spec_autocorr,
-34
+                                      get_antenna_selection_include_autocorr,
-35
+                                      table_manager)
-36
+import casatasks.private.task_sdatmcor as sdatmcor_impl
-495
-496
+    def test_custom_atm_params_non_conform_list_input(self):
-497
+        """Test customized ATM parameters: non-conform layerboundaries and layertemperature."""
-498
+        with self.assertRaises(Exception):
-499
+            sdatmcor(
-500
+                infile=self.infile, outfile=self.outfile, datacolumn='data',
-501
+                atmdetail=True,
-502
+                layerboundaries='800m,1.5km', layertemperature='250K,200K,190K'
-503
+            )
-504
-+
+    def test_omp_num_threads(self):
-+
+        """Test if the task respects OMP_NUM_THREADS environment variable."""
-+
+        omp_num_threads_org = os.environ.get('OMP_NUM_THREADS')
-+
+        try:
-+
+            # set num_threads for OpenMP to any different value than the current one
-+
+            if omp_num_threads_org is None:
-+
+                num_threads = 2
-+
+            else:
-+
+                iter = filter(lambda x: x != int(omp_num_threads_org), range(2, 9))
-+
+                num_threads = next(iter)
-+
+                self.assertNotEqual(num_threads, omp_num_threads_org)
-+
+            os.environ['OMP_NUM_THREADS'] = f'{num_threads}'
-+
-+
+            # run task
-+
+            sdatmcor(infile=self.infile, outfile=self.outfile, datacolumn='data')
-+
+        finally:
-+
+            if omp_num_threads_org is None:
-+
+                os.environ.pop('OMP_NUM_THREADS')
-+
+            else:
-+
+                os.environ['OMP_NUM_THREADS'] = omp_num_threads_org
-+
-+
+        # consistency check
-+
+        if omp_num_threads_org is None:
-+
+            self.assertIsNone(os.environ.get('OMP_NUM_THREADS'))
-+
+        else:
-+
+            self.assertEqual(os.environ.get('OMP_NUM_THREADS'), omp_num_threads_org)
-+
-+
+        # check log
-+
+        if os.path.exists(casalog.logfile()):
-+
+            with open(casalog.logfile(), 'r') as f:
-+
+                pattern = re.compile(r'.*Setting numThreads_ to ([0-9+])')
-+
+                lines = list(filter(lambda x: x is not None, map(lambda x: re.search(pattern, x), f)))
-+
+            num_threads_log = int(lines[-1].group(1))
-+
-+
+            print(f'{OMP_NUM_THREADS_INITIAL} {omp_num_threads_org} {num_threads} {num_threads_log}')
-+
+            self.assertEqual(num_threads, num_threads_log)
-+
-+
+        # check output MS
-+
+        self.check_result({19: True, 23: True})
-+
-545
-546
+class ATMParamTest(unittest.TestCase):
-547
+    def _param_test_template(self, valid_test_cases,
-548
+                             invalid_user_input, user_default, task_default, unit=''):
-549
+        # internal error
-550
+        wrong_task_default = 'NG'
-551
+        with self.assertRaises(RuntimeError):
-552
+            param, is_customized = sdatmcor_impl.parse_atm_params(
-553
+                '',
-554
+                user_default,

casa6

Commits

Jira Issues

Add shortcut