use np.allclose

modflowpy · Nov 21, 2023 · 88b9584 · 88b9584
1 parent 61b66d9
commit 88b9584
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/autotest/test_plotutil.py b/autotest/test_plotutil.py
@@ -304,10 +304,4 @@ def test_to_prt_pathlines_roundtrip():
     prt_pls.drop(
         ["imdl", "iprp", "name", "istatus", "ireason"], axis=1, inplace=True
     )
-    inp_pls[inp_pls.select_dtypes(np.float64).columns] = inp_pls.select_dtypes(
-        np.float64
-    ).astype(np.float32)
-    inp_pls[inp_pls.select_dtypes(np.int64).columns] = inp_pls.select_dtypes(
-        np.int64
-    ).astype(np.int32)
-    assert np.array_equal(inp_pls, prt_pls)
+    assert np.allclose(inp_pls, prt_pls)