[System Tests] Change Spark tests to use a single window (#1872)

mlrun · Apr 6, 2022 · 325b44a · 325b44a
1 parent da9d526
commit 325b44a
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/tests/system/feature_store/test_spark_engine.py b/tests/system/feature_store/test_spark_engine.py
@@ -350,7 +350,7 @@ def test_aggregations_emit_every_event(self):
         data_set.add_aggregation(
             column="bid",
             operations=["sum", "max", "count"],
-            windows=["1h", "2h"],
+            windows=["2h"],
             period="10m",
             emit_policy=EmitEveryEvent(),
         )
@@ -368,9 +368,6 @@ def test_aggregations_emit_every_event(self):
         expected_results = df.to_dict(orient="list")
         expected_results.update(
             {
-                "bid_sum_1h": [2000, 10, 12, 26, 24],
-                "bid_max_1h": [2000, 10, 12, 16, 16],
-                "bid_count_1h": [1, 1, 1, 2, 2],
                 "bid_sum_2h": [2000, 10, 2012, 26, 34],
                 "bid_max_2h": [2000, 10, 2000, 16, 16],
                 "bid_count_2h": [1, 1, 2, 2, 3],
@@ -389,7 +386,7 @@ def test_aggregations_emit_every_event(self):
         storey_data_set.add_aggregation(
             column="bid",
             operations=["sum", "max", "count"],
-            windows=["1h", "2h"],
+            windows=["2h"],
             period="10m",
         )
         fs.ingest(storey_data_set, source)