Fix typo and add asserts.

pnnl · Sep 23, 2024 · 8ca2e30 · 8ca2e30
1 parent 2853446
commit 8ca2e30
Showing 1 changed file with 5 additions and 1 deletion.
diff --git 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