diff --git a/tests/parametric_test.py b/tests/parametric_test.py
index 07a5422..872532b 100644
--- a/tests/parametric_test.py
+++ b/tests/parametric_test.py
@@ -55,6 +55,9 @@ def assert_function_1(_, files):
 
 
 def assert_function_2(output, _):
+    # np.float64(5.179525402166653)\n
+    if "np.float64" in output.content:
+        return 4.0 <= float(output.content.split("(")[1].split(")")[0]) <= 7.0
     return 4.0 <= float(output.content) <= 7.0