more tests for the normalizing function

paulalndwhr · Oct 14, 2023 · 09c847b · 09c847b
1 parent 2995ba6
commit 09c847b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_functions.py b/tests/test_functions.py
@@ -14,7 +14,10 @@ def f(x): return 1
 @pytest.mark.parametrize("test_function,test_domain,argument,expected", [
     (uniform, [0, 1], 0, 1.0),
     (uniform, [0, 1], 0.5, 1.0),
-    (uniform, [0, 1], 1, 1.0)
+    (uniform, [0, 1], 1, 1.0),
+    (lambda x: x, [0, 1], 0, 0),
+    (lambda x: x, [0, 1], 0.5, 1),
+    (lambda x: x, [0, 1], 1, 2),
 ])
 def test_normalize(test_function, test_domain, argument, expected):
     assert normalize(test_function, test_domain)(argument) == expected