relax dry drop comparison

wpbonelli · Dec 5, 2024 · 83e6671 · 83e6671
1 parent 2134a97
commit 83e6671
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/autotest/__snapshots__/test_prt_dry/test_mf6model[2-prtdry_drop].npy b/autotest/__snapshots__/test_prt_dry/test_mf6model[2-prtdry_drop].npy
diff --git a/autotest/test_prt_dry.py b/autotest/test_prt_dry.py
@@ -372,7 +372,8 @@ def check_output(idx, test, snapshot):
     strtpts = pls[pls.ireason == 0]
 
     # compare to expected results
-    actual = pls.drop(["name", "icell"], axis=1).round(2).reset_index(drop=True)
+    decimals = 1 if "drop" in name else 2
+    actual = pls.drop(["name", "icell"], axis=1).round(decimals).reset_index(drop=True)
     assert snapshot == actual.to_records(index=False)
 
     plot_pathlines = False