add test_rejection

wrote a test to check wheter the rejection class works as intendet

Author: daklauss

Diff for: pyproject.toml

@@ -72,4 +72,4 @@ indent-width = 4
 
 [tool.ruff.lint]
 select = ["E", "F", "W"]
-ignore = ["F401"]
+ignore = ["F401"]

Diff for: tests/test_rejection.py

@@ -0,0 +1,37 @@
+import numpy as np
+import pytest
+
+from CADETPythonSimulator.rejection import (
+    RejectionBase, StepCutOff
+)
+
+
+
+
+
+
+
+
+TestCaseCutof = {
+            "model": StepCutOff,
+            "model_param": {
+                "cutoff_weight": 0.5
+            },
+            "weights": [0, 0.5, 1],
+            "expected": [0, 1, 1]
+        }
+
+@pytest.mark.parametrize(
+    "parameters",
+    [
+        TestCaseCutof
+    ]
+)
+
+class TestRejection():
+    def test_get_rejection(self, parameters):
+
+        model = parameters["model"](**parameters["model_param"])
+
+        solution = [model.get_rejection(weight) for weight in parameters["weights"]]
+        np.testing.assert_array_almost_equal(solution, parameters["expected"])

Diff for: tests/test_residual.py

@@ -96,7 +96,10 @@ def test_calculation_concentration_cstr(self, parameters):
 
         param_vec_conc = parameters["values"].values()
 
-        np.testing.assert_array_almost_equal(calculate_residual_concentration_cstr(*param_vec_conc), parameters["expected"])
+        np.testing.assert_array_almost_equal(
+            calculate_residual_concentration_cstr(*param_vec_conc),
+            parameters["expected"]
+        )
 
 
 

Diff for: tests/test_unit_operation.py

@@ -316,7 +316,7 @@ def test_set_inlet_state(self, unit_operation: UnitOperationBase, expected: dict
         }
         if unit_operation.n_inlet_ports == 0:
             with pytest.raises(Exception):
-                unit_operation.set_inlet_port_state(s_in, 0)
+                unit_operation.set_inlet_state_flat(s_in, 0)
         else:
             for port in range(unit_operation.n_inlet_ports):
                 unit_operation.set_inlet_state_flat(s_in, port)