QT/superset2
1
0
Fork 0
mirror of https://github.com/apache/superset.git synced 2026-04-20 00:24:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add resample operator in post processing (#16607)

Browse Source 
* feat: add resample operator in post processing

* wip

* fill zero values

* updates

* fix ut
This commit is contained in:
Yongjie Zhao
2021-09-17 09:40:59 +01:00
committed by GitHub
parent a4f60010a1
commit cc1c6c1bb6
3 changed files with 46 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
tests/integration_tests/pandas_postprocessing_tests.py
View File

@@ -870,3 +870,22 @@ class TestPostProcessing(SupersetTestCase):
metrics=["cars"],
percentiles=[10, 90, 10],
)
def test_resample(self):
df = timeseries_df.copy()
df.index.name = "time_column"
df.reset_index(inplace=True)
post_df = proc.resample(
df=df, rule="1D", method="ffill", time_column="time_column",
)
self.assertListEqual(
post_df["label"].tolist(), ["x", "y", "y", "y", "z", "z", "q"]
)
self.assertListEqual(post_df["y"].tolist(), [1.0, 2.0, 2.0, 2.0, 3.0, 3.0, 4.0])
post_df = proc.resample(
df=df, rule="1D", method="asfreq", time_column="time_column", fill_value=0,
)
self.assertListEqual(post_df["label"].tolist(), ["x", "y", 0, 0, "z", 0, "q"])
self.assertListEqual(post_df["y"].tolist(), [1.0, 2.0, 0, 0, 3.0, 0, 4.0])