fix: style

quintoandar · Sep 16, 2024 · 11a2973 · 11a2973
1 parent c4c1e5b
commit 11a2973
Showing 1 changed file with 30 additions and 5 deletions.
diff --git a/tests/unit/butterfree/transform/conftest.py b/tests/unit/butterfree/transform/conftest.py
@@ -92,11 +92,36 @@ def make_output_filtering_dataframe(spark_context, spark_session):
 
 def make_rolling_windows_agg_dataframe(spark_context, spark_session):
     data = [
-        {"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},
+        {
+            "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")