write tests for bound_window and bound_range

Author: cadenmyers13

tests/test_parameter.py

@@ -16,6 +16,9 @@
 
 import unittest
 
+import numpy as np
+import pytest
+
 from diffpy.srfit.fitbase.parameter import (
     Parameter,
     ParameterAdapter,
@@ -121,5 +124,95 @@ def testWrapper(self):
         return
 
 
+@pytest.mark.parametrize(
+    "lower, upper, expected",
+    [
+        # User sets both lower and upper bounds explicitly.
+        (1, 10, [1, 10]),
+        # User sets only a lower bound.
+        (2, None, [2, np.inf]),
+        # User sets only an upper bound.
+        (None, 8, [-np.inf, 8]),
+        # User overwrites existing bounds.
+        (2, 6, [2, 6]),
+    ],
+)
+def test_bound_range(lower, upper, expected):
+    p = Parameter("a", value=5)
+    # If testing overwrite, pre-set bounds to see overwrite effect
+    if expected == [2, 6]:
+        p.bound_range(0, 10)
+    p.bound_range(lower_bound=lower, upper_bound=upper)
+    actual = p.bounds
+    assert actual == expected
+
+
+@pytest.mark.parametrize(
+    "lower, upper, expected",
+    [
+        # User sets both lower and upper bounds explicitly.
+        (1, 10, [1, 10]),
+        # User sets only a lower bound.
+        (2, None, [2, np.inf]),
+        # User sets only an upper bound.
+        (None, 8, [-np.inf, 8]),
+        # User overwrites existing bounds.
+        (2, 6, [2, 6]),
+    ],
+)
+def test_boundRange(lower, upper, expected):
+    p = Parameter("a", value=5)
+    # If testing overwrite, pre-set bounds to see overwrite effect
+    if expected == [2, 6]:
+        p.boundRange(0, 10)
+    p.boundRange(lower_bound=lower, upper_bound=upper)
+    actual = p.bounds
+    assert actual == expected
+
+
+@pytest.mark.parametrize(
+    "value, lower_radius, upper_radius, expected",
+    [
+        # Symmetric radius (upper_radius None, uses lower_radius)
+        (10, 2, None, [8, 12]),
+        # Asymmetric radius
+        (10, 3, 5, [7, 15]),
+        # Zero radius
+        (4, 0, None, [4, 4]),
+        # Current value updated before bounding
+        (20, 2, None, [18, 22]),
+    ],
+)
+def test_bound_window(value, lower_radius, upper_radius, expected):
+    p = Parameter("a", value=5)
+    if value != 5:
+        p.set_value(value)
+    p.bound_window(lower_radius=lower_radius, upper_radius=upper_radius)
+    actual = p.bounds
+    assert actual == expected
+
+
+@pytest.mark.parametrize(
+    "value, lower_radius, upper_radius, expected",
+    [
+        # Symmetric radius (upper_radius None, uses lower_radius)
+        (10, 2, None, [8, 12]),
+        # Asymmetric radius
+        (10, 3, 5, [7, 15]),
+        # Zero radius
+        (4, 0, None, [4, 4]),
+        # Current value updated before bounding
+        (20, 2, None, [18, 22]),
+    ],
+)
+def test_boundWindow(value, lower_radius, upper_radius, expected):
+    p = Parameter("a", value=5)
+    if value != 5:
+        p.set_value(value)
+    p.boundWindow(lr=lower_radius, ur=upper_radius)
+    actual = p.bounds
+    assert actual == expected
+
+
 if __name__ == "__main__":
     unittest.main()