Update tests

Author: henrikjacobsenfys

src/easydynamics/sample_model/components/polynomial.py

@@ -66,6 +66,13 @@ def evaluate(self, x: Union[Numeric, sc.Variable]) -> np.ndarray:
                 )
         else:
             x_in = x
+
+        if any(np.isnan(x_in)):
+            raise ValueError("Input x contains NaN values.")
+
+        if any(np.isinf(x_in)):
+            raise ValueError("Input x contains infinite values.")
+
         result = np.zeros_like(x_in, dtype=float)
         for i, param in enumerate(self.coefficients):
             result += param.value * np.power(x_in, i)

tests/unit_tests/sample_model/components/test_damped_harmonic_oscillator.py

@@ -99,7 +99,7 @@ def test_area_property_setter(self, dho: DampedHarmonicOscillator):
 
     def test_center_property_getter(self, dho: DampedHarmonicOscillator):
         # WHEN THEN EXPECT
-        assert dho.center.value == 0.5
+        assert dho.center.value == 1.5
 
     def test_center_property_setter(self, dho: DampedHarmonicOscillator):
         # WHEN
@@ -111,7 +111,7 @@ def test_center_property_setter(self, dho: DampedHarmonicOscillator):
             dho.center = "invalid"
 
     def test_width_property_getter(self, dho: DampedHarmonicOscillator):
-        assert dho.width.value == 0.6
+        assert dho.width.value == 0.3
 
     def test_width_property_setter(self, dho: DampedHarmonicOscillator):
         # WHEN
@@ -170,6 +170,22 @@ def test_evaluate_with_incompatible_unit(self, dho: DampedHarmonicOscillator):
         ):
             dho.evaluate(x)
 
+    def test_evaluate_with_nan_input(self, dho: DampedHarmonicOscillator):
+        # WHEN
+        x = np.array([0.0, np.nan, 1.0])
+
+        # THEN EXPECT
+        with pytest.raises(ValueError, match="Input x contains NaN values."):
+            dho.evaluate(x)
+
+    def test_evaluate_with_infinite_input(self, dho: DampedHarmonicOscillator):
+        # WHEN
+        x = np.array([0.0, np.inf, 1.0])
+
+        # THEN EXPECT
+        with pytest.raises(ValueError, match="Input x contains infinite values."):
+            dho.evaluate(x)
+
     def test_get_parameters(self, dho: DampedHarmonicOscillator):
         params = dho.get_parameters()
         assert len(params) == 3

tests/unit_tests/sample_model/components/test_gaussian.py

@@ -138,6 +138,22 @@ def test_evaluate_with_incompatible_unit(self, gaussian: Gaussian):
         ):
             gaussian.evaluate(x)
 
+    def test_evaluate_with_nan_input(self, gaussian: Gaussian):
+        # WHEN
+        x = np.array([0.0, np.nan, 1.0])
+
+        # THEN EXPECT
+        with pytest.raises(ValueError, match="Input x contains NaN values."):
+            gaussian.evaluate(x)
+
+    def test_evaluate_with_infinite_input(self, gaussian: Gaussian):
+        # WHEN
+        x = np.array([0.0, np.inf, 1.0])
+
+        # THEN EXPECT
+        with pytest.raises(ValueError, match="Input x contains infinite values."):
+            gaussian.evaluate(x)
+
     def test_center_is_fixed_if_set_to_None(self):
         # WHEN
         test_gaussian = Gaussian(

tests/unit_tests/sample_model/components/test_lorentzian.py

@@ -140,6 +140,24 @@ def test_evaluate_with_incompatible_unit(self, lorentzian: Lorentzian):
         ):
             lorentzian.evaluate(x)
 
+    def test_evaluate_with_nan_input(self, lorentzian: Lorentzian):
+        # WHEN
+        x = np.array([0.0, np.nan, 1.0])
+
+        # THEN EXPECT
+        with pytest.raises(ValueError, match="Input x contains NaN values."):
+            lorentzian.evaluate(x)
+
+    def test_evaluate_with_infinite_input(self, lorentzian: Lorentzian):
+        # WHEN
+        x = np.array([0.0, np.inf, 1.0])
+
+        # THEN EXPECT
+        with pytest.raises(ValueError, match="Input x contains infinite values."):
+            lorentzian.evaluate(x)
+
+            lorentzian.evaluate(x)
+
     def test_center_is_fixed_if_set_to_None(self):
         # WHEN
         test_lorentzian = Lorentzian(

tests/unit_tests/sample_model/components/test_polynomial.py

@@ -75,6 +75,22 @@ def test_evaluate_with_different_unit_error(self, polynomial: Polynomial):
         ):
             polynomial.evaluate(x)
 
+    def test_evaluate_with_nan_input(self, polynomial: Polynomial):
+        # WHEN
+        x = np.array([0.0, np.nan, 1.0])
+
+        # THEN EXPECT
+        with pytest.raises(ValueError, match="Input x contains NaN values."):
+            polynomial.evaluate(x)
+
+    def test_evaluate_with_infinite_input(self, polynomial: Polynomial):
+        # WHEN
+        x = np.array([0.0, np.inf, 1.0])
+
+        # THEN EXPECT
+        with pytest.raises(ValueError, match="Input x contains infinite values."):
+            polynomial.evaluate(x)
+
     def test_degree(self, polynomial: Polynomial):
         # WHEN THEN EXPECT
         assert polynomial.degree() == 2

tests/unit_tests/sample_model/components/test_voigt.py

@@ -219,6 +219,22 @@ def test_evaluate_with_incompatible_unit(self, voigt: Voigt):
         ):
             voigt.evaluate(x)
 
+    def test_evaluate_with_nan_input(self, voigt: Voigt):
+        # WHEN
+        x = np.array([0.0, np.nan, 1.0])
+
+        # THEN EXPECT
+        with pytest.raises(ValueError, match="Input x contains NaN values."):
+            voigt.evaluate(x)
+
+    def test_evaluate_with_infinite_input(self, voigt: Voigt):
+        # WHEN
+        x = np.array([0.0, np.inf, 1.0])
+
+        # THEN EXPECT
+        with pytest.raises(ValueError, match="Input x contains infinite values."):
+            voigt.evaluate(x)
+
     def test_center_is_fixed_if_set_to_None(self):
         # WHEN THEN
         test_voigt = Voigt(