fix: rolling and cum operator on multiple series (#16945)

* fix: rolling and cum operator on multiple series

* add UT

* updates

(cherry picked from commit fd8461406d)
This commit is contained in:
Yongjie Zhao
2021-10-07 09:42:08 +01:00
committed by Elizabeth Thompson
parent 4d33f7b7b6
commit 19d2fef490
3 changed files with 185 additions and 9 deletions

View File

@@ -165,3 +165,19 @@ prophet_df = DataFrame(
"b": [4, 3, 4.1, 3.95],
}
)
single_metric_df = DataFrame(
{
"dttm": to_datetime(["2019-01-01", "2019-01-01", "2019-01-02", "2019-01-02",]),
"country": ["UK", "US", "UK", "US"],
"sum_metric": [5, 6, 7, 8],
}
)
multiple_metrics_df = DataFrame(
{
"dttm": to_datetime(["2019-01-01", "2019-01-01", "2019-01-02", "2019-01-02",]),
"country": ["UK", "US", "UK", "US"],
"sum_metric": [5, 6, 7, 8],
"count_metric": [1, 2, 3, 4],
}
)