more tests of args

Author: martinkilbinger

cs_util/tests/test_args.py

@@ -12,6 +12,7 @@
 import pytest
 
 from unittest import TestCase
+from unittest.mock import patch
 
 from cs_util import args
 
@@ -51,13 +52,17 @@ def setUp(self):
             "p_Bool": "bool option, set to True if given",
             "p_str": "string option, default={}",
         }
+        self._string0 = None
+        self._string1 = "abc"
+        self._string3 = "a b c"
 
     def tearDown(self):
         """Unset test parameter values."""
         self._params_def = None
         self._types = None
-
+        self._short_options = None
         self._help_strings = None
+        self._string1 = None
 
     def test_parse_options(self):
         """Test `cs_util.args.parse_options` method."""
@@ -79,4 +84,47 @@ def test_parse_options(self):
         npt.assert_equal(self._options["p_Bool"], False)
         npt.assert_equal(self._options["p_str_def"], "second_string")
 
-        # Test exceptions TBD
+        # Test default args array
+        test_argv = [None, "-i", "2"]
+        with patch("sys.argv", test_argv):
+            self._options = args.parse_options(
+                self._params_def,
+                self._short_options,
+                self._types,
+                self._help_strings,
+                args=None,
+            )
+            npt.assert_equal(self._options["p_int"], 2)
+
+    def test_my_string_split(self):
+        """Test `cs_util.args.my_string_split` method."""
+
+        # Test string=None
+        results = args.my_string_split(self._string0)
+        self.assertIsNone(results)
+
+        # Test string without separator
+        results = args.my_string_split(self._string1)
+        npt.assert_equal(len(results), 1)
+        npt.assert_equal(results[0], self._string1)
+
+        # Test string with separators
+        results = args.my_string_split(self._string3)
+        npt.assert_equal(len(results), 3)
+        for idx in (0, 1, 2):
+            npt.assert_equal(results[idx], self._string3[idx * 2])
+
+        # Test different separator
+        results = args.my_string_split(self._string3, sep="_")
+        npt.assert_equal(len(results), 1)
+        npt.assert_equal(results[0], self._string3)
+
+        # Test mismatching number of substrings
+        # Exception for stop=True
+        with self.assertRaises(ValueError):
+            args.my_string_split(self._string3, num=2, stop=True)
+        # Return value when stop is False
+        results = args.my_string_split(self._string3, num=2, stop=False)
+        npt.assert_equal(len(results), 3)
+        for idx in (0, 1, 2):
+            npt.assert_equal(results[idx], self._string3[idx * 2])

cs_util/tests/test_cosmo.py

@@ -27,7 +27,7 @@ def setUp(self):
         self._z_lens = 0.5
         self._z_source_arr = [0.4, 0.6, 0.8, 0.9]
         self._nz_source_arr = [0.5, 0.6, 2.2, 1.6]
-        self._cosmo = ccl.core.CosmologyVanillaLCDM()
+        self._cosmo = ccl.CosmologyVanillaLCDM()
         self._sigma_crit_value = 3920.1478
         # Value verified with package dsigma as 3919.700
 
@@ -87,7 +87,7 @@ def test_sigma_crit(self):
         npt.assert_almost_equal(
             sigma_crit.value,
             self._sigma_crit_value,
-            decimal=4,
+            decimal=3,
         )
         # Test return unit
         npt.assert_equal(sigma_crit.unit, self._sigma_crit_unit)
@@ -148,7 +148,7 @@ def test_sigma_crit_eff(self):
         npt.assert_almost_equal(
             sigma_crit_eff.value,
             self._sigma_crit_value_eff,
-            decimal=4,
+            decimal=3,
         )
 
         # Test return unit