diff --git a/tests/test_strip.py b/tests/test_strip.py index 0f1ad5ee..16c310c4 100644 --- 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"]