From c4c1e5b6bf476f8d8484f6725a32914d1c51b9c2 Mon Sep 17 00:00:00 2001 From: Ralph Filho Date: Mon, 16 Sep 2024 14:12:42 -0300 Subject: [PATCH] fix: test --- tests/unit/butterfree/transform/conftest.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/unit/butterfree/transform/conftest.py b/tests/unit/butterfree/transform/conftest.py index 07ab44f2..fc16cc57 100644 --- a/tests/unit/butterfree/transform/conftest.py +++ b/tests/unit/butterfree/transform/conftest.py @@ -92,11 +92,11 @@ def make_output_filtering_dataframe(spark_context, spark_session): def make_rolling_windows_agg_dataframe(spark_context, spark_session): data = [ - (1, "2016-04-11 00:00:00", [None, None], None), - (1, "2016-04-12 00:00:00", [300.0, 350.0], None), - (1, "2016-04-19 00:00:00", [None, None], None), - (1, "2016-04-23 00:00:00", [1000.0, 1100.0], None), - (1, "2016-04-30 00:00:00", [None, None], None), + {"id": 1, "timestamp": "2016-04-11 00:00:00", "feature1__avg_over_1_week_rolling_windows": None, "feature2__avg_over_1_week_rolling_windows": None}, + {"id": 1, "timestamp": "2016-04-12 00:00:00", "feature1__avg_over_1_week_rolling_windows": 300.0, "feature2__avg_over_1_week_rolling_windows": 350.0}, + {"id": 1, "timestamp": "2016-04-19 00:00:00", "feature1__avg_over_1_week_rolling_windows": None, "feature2__avg_over_1_week_rolling_windows": None}, + {"id": 1, "timestamp": "2016-04-23 00:00:00", "feature1__avg_over_1_week_rolling_windows": 1000.0, "feature2__avg_over_1_week_rolling_windows": 1100.0}, + {"id": 1, "timestamp": "2016-04-30 00:00:00", "feature1__avg_over_1_week_rolling_windows": None, "feature2__avg_over_1_week_rolling_windows": None}, ] return create_dataframe(data, timestamp_col="timestamp")