tweak test for to_dataframe change

pmlmodelling · Nov 29, 2024 · 7843b2e · 7843b2e
1 parent 47e9432
commit 7843b2e
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/test_strip.py b/tests/test_strip.py
@@ -19,10 +19,10 @@ def test_empty(self):
         ds.top()
         ds.subset(time = 0)
         ds.spatial_mean()
-        df1 = ds.to_dataframe()
+        df1 = ds.to_dataframe(drop_bnds = False)
         ds.strip_variables()
         assert 'ncks -C -v t_an,t_dd,t_gp,t_ma,t_mn,t_oa,t_sd,t_se' in ds.history[1]
-        df2 = ds.to_dataframe()
+        df2 = ds.to_dataframe(drop_bnds = False)
         assert len([x for x in df1.columns if "bnds" in x]) > 0
         assert len([x for x in df2.columns if "bnds" in x]) == 0
 
@@ -32,9 +32,9 @@ def test_empty(self):
         ds.top()
         ds.subset(time = 0)
         ds.spatial_mean()
-        df1 = ds.to_dataframe()
+        df1 = ds.to_dataframe(drop_bnds = False)
         ds.strip_variables("t_an")
-        df2 = ds.to_dataframe()
+        df2 = ds.to_dataframe(drop_bnds = False)
         assert len([x for x in df1.columns if "bnds" in x]) > 0
         assert len([x for x in df2.columns if "bnds" in x]) == 0
         assert ds.variables == ["t_an"]