add test_rejection

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

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"])

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"]
+        )