fix: test

quintoandar · Sep 16, 2024 · c4c1e5b · c4c1e5b
1 parent 3d52fa0
commit c4c1e5b
Showing 1 changed file with 5 additions and 5 deletions.
diff --git 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")