python: fix failing tests

tweag · Feb 13, 2025 · 10d6861 · 10d6861
1 parent 05d865e
commit 10d6861
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/python/native/test.py b/examples/python/native/test.py
@@ -12,7 +12,7 @@ def test_nb(tb):
     assert sp_result == "[ 0.5  2.5  6.5 12.5]"
 
     mpl_result = tb.cell_output_text(3)
-    assert mpl_result == "(0.122312, 0.633153, 0.530398, 1.0)"
+    assert mpl_result == "(np.float64(0.122312), np.float64(0.633153), np.float64(0.530398), np.float64(1.0))"
 
 if __name__ == '__main__':
     test_nb()

diff --git a/examples/python/science/test.py b/examples/python/science/test.py
@@ -12,7 +12,7 @@ def test_nb(tb):
     assert sp_result == "[ 0.5  2.5  6.5 12.5]"
 
     mpl_result = tb.cell_output_text(3)
-    assert mpl_result == "(0.122312, 0.633153, 0.530398, 1.0)"
+    assert mpl_result == "(np.float64(0.122312), np.float64(0.633153), np.float64(0.530398), np.float64(1.0))"
 
 if __name__ == '__main__':
     test_nb()