updated tests

franciscojavierarceo · franciscojavierarceo · commit 86dc577bb7e7 · 2025-09-23T10:49:57.000+01:00
Signed-off-by: Francisco Javier Arceo &lt;farceo@redhat.com&gt;
diff --git a/sdk/python/feast/feature_server.py b/sdk/python/feast/feature_server.py
@@ -438,11 +438,15 @@ def materialize(request: MaterializeRequest) -> None:
         if request.disable_event_timestamp:
             # Query all available data and use current datetime as event timestamp
             now = datetime.now()
-            start_date = datetime(1970, 1, 1)  # Beginning of time to capture all historical data
+            start_date = datetime(
+                1970, 1, 1
+            )  # Beginning of time to capture all historical data
             end_date = now
         else:
             if not request.start_ts or not request.end_ts:
-                raise ValueError("start_ts and end_ts are required when disable_event_timestamp is False")
+                raise ValueError(
+                    "start_ts and end_ts are required when disable_event_timestamp is False"
+                )
             start_date = utils.make_tzaware(parser.parse(request.start_ts))
             end_date = utils.make_tzaware(parser.parse(request.end_ts))
 
diff --git a/sdk/python/tests/unit/cli/test_cli.py b/sdk/python/tests/unit/cli/test_cli.py
@@ -208,13 +208,22 @@ def test_cli_materialize_disable_event_timestamp():
 
         # Test that providing timestamps with --disable-event-timestamp fails
         return_code, output = runner.run_with_output(
-            ["materialize", "--disable-event-timestamp", "2021-01-01T00:00:00", "2021-01-02T00:00:00"],
-            cwd=repo_path
+            [
+                "materialize",
+                "--disable-event-timestamp",
+                "2021-01-01T00:00:00",
+                "2021-01-02T00:00:00",
+            ],
+            cwd=repo_path,
         )
         assertpy.assert_that(return_code).is_equal_to(2)  # Click usage error
-        assertpy.assert_that(output).contains(b"Cannot specify START_TS or END_TS when --disable-event-timestamp is used")
+        assertpy.assert_that(output).contains(
+            b"Cannot specify START_TS or END_TS when --disable-event-timestamp is used"
+        )
 
         # Test that missing timestamps without flag fails
         return_code, output = runner.run_with_output(["materialize"], cwd=repo_path)
         assertpy.assert_that(return_code).is_equal_to(2)  # Click usage error
-        assertpy.assert_that(output).contains(b"START_TS and END_TS are required unless --disable-event-timestamp is used")
+        assertpy.assert_that(output).contains(
+            b"START_TS and END_TS are required unless --disable-event-timestamp is used"
+        )
diff --git a/sdk/python/tests/unit/test_feature_server.py b/sdk/python/tests/unit/test_feature_server.py
@@ -155,8 +155,8 @@ def test_materialize_with_timestamps(test_client):
         json={
             "start_ts": "2021-01-01T00:00:00",
             "end_ts": "2021-01-02T00:00:00",
-            "feature_views": ["driver_hourly_stats"]
-        }
+            "feature_views": ["driver_hourly_stats"],
+        },
     )
     assert response.status_code == 200
 
@@ -167,19 +167,16 @@ def test_materialize_disable_event_timestamp(test_client):
         "/materialize",
         json={
             "feature_views": ["driver_hourly_stats"],
-            "disable_event_timestamp": True
-        }
+            "disable_event_timestamp": True,
+        },
     )
     assert response.status_code == 200
 
 
 def test_materialize_missing_timestamps_fails(test_client):
     """Test that missing timestamps without disable_event_timestamp fails"""
     response = test_client.post(
-        "/materialize",
-        json={
-            "feature_views": ["driver_hourly_stats"]
-        }
+        "/materialize", json={"feature_views": ["driver_hourly_stats"]}
     )
     assert response.status_code == 422  # Validation error for missing required fields
 
@@ -189,10 +186,7 @@ def test_materialize_request_model():
     from feast.feature_server import MaterializeRequest
 
     # Test with disable_event_timestamp=True (no timestamps needed)
-    req1 = MaterializeRequest(
-        feature_views=["test"],
-        disable_event_timestamp=True
-    )
+    req1 = MaterializeRequest(feature_views=["test"], disable_event_timestamp=True)
     assert req1.disable_event_timestamp is True
     assert req1.start_ts is None
     assert req1.end_ts is None
@@ -201,7 +195,7 @@ def test_materialize_request_model():
     req2 = MaterializeRequest(
         start_ts="2021-01-01T00:00:00",
         end_ts="2021-01-02T00:00:00",
-        feature_views=["test"]
+        feature_views=["test"],
     )
     assert req2.disable_event_timestamp is False
     assert req2.start_ts == "2021-01-01T00:00:00"
