Add a props parameter to dataframe and default to grabbing all non-sy…

…stem props
aazuspan · Aug 27, 2022 · 63bb869 · 63bb869
1 parent 47368f6
commit 63bb869
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/test/test_time_series.py b/test/test_time_series.py
@@ -444,8 +444,10 @@ def test_dataframe():
  for i in range(3)
  ]
  ts = wxee.TimeSeries(imgs).set("system:id", col_id)
- df = ts.dataframe()
+ df = ts.dataframe(["system:time_start", "system:id"])
 
  assert df.index.id == col_id
- assert df.time_start.dt.strftime("%Y-%m-%d").values.tolist() == start_dates
- assert df.id.values.tolist() == ids
+ assert (
+ df["system:time_start"].dt.strftime("%Y-%m-%d").values.tolist() == start_dates
+ )
+ assert df["system:id"].values.tolist() == ids