From 8ca2e30acf5b17a73a688371ddf455af383d90de Mon Sep 17 00:00:00 2001 From: "Lerond, Jeremy" Date: Mon, 23 Sep 2024 13:33:34 -0700 Subject: [PATCH] Fix typo and add asserts. --- tests/test_unitarydirectexpansion.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/test_unitarydirectexpansion.py b/tests/test_unitarydirectexpansion.py index ad7b96c..614f702 100644 --- a/tests/test_unitarydirectexpansion.py +++ b/tests/test_unitarydirectexpansion.py @@ -340,4 +340,8 @@ def test_NN_wght_avg(self): assert len(set_of_curves) == 5 # Check normalization - assert round(set_of_curves.curves[0].evaluate(19, 35), 2) == 1.0 + assert round(set_of_curves[0].evaluate(19.44, 35), 2) == 1.0 + assert round(set_of_curves[1].evaluate(19.44, 35), 2) == 1.0 + assert round(set_of_curves[2].evaluate(1.0, 0), 2) == 1.0 + assert round(set_of_curves[3].evaluate(1.0, 0), 2) == 1.0 + assert round(set_of_curves[4].evaluate(1.0, 0), 2) == 1.0