test fix

bubriks · bubriks · commit c61fc75706d4 · 2024-10-08T10:19:12.000+03:00
diff --git a/python/tests/core/test_feature_group_engine.py b/python/tests/core/test_feature_group_engine.py
@@ -547,7 +547,7 @@ def test_clean_delta(self, mocker):
         )
 
         # Act
-        fg_engine.clean(feature_group=fg, write_options={})
+        fg_engine.delta_vacuum(feature_group=fg, retention_hours=200)
 
         # Assert
         assert mock_hudi_engine.return_value.vacuum.call_count == 1
@@ -571,7 +571,7 @@ def test_clean_hudi(self, mocker):
         )
 
         # Act
-        fg_engine.clean(feature_group=fg, write_options={})
+        fg_engine.delta_vacuum(feature_group=fg, retention_hours=200)
 
     def test_sql(self, mocker):
         # Arrange

Original file line number	Diff line number	Diff line change
`@@ -547,7 +547,7 @@ def test_clean_delta(self, mocker):`
`547`	`547`	`)`
`548`	`548`
`549`	`549`	`# Act`
`550`		`- fg_engine.clean(feature_group=fg, write_options={})`
	`550`	`+ fg_engine.delta_vacuum(feature_group=fg, retention_hours=200)`
`551`	`551`
`552`	`552`	`# Assert`
`553`	`553`	`assert mock_hudi_engine.return_value.vacuum.call_count == 1`
`@@ -571,7 +571,7 @@ def test_clean_hudi(self, mocker):`
`571`	`571`	`)`
`572`	`572`
`573`	`573`	`# Act`
`574`		`- fg_engine.clean(feature_group=fg, write_options={})`
	`574`	`+ fg_engine.delta_vacuum(feature_group=fg, retention_hours=200)`
`575`	`575`
`576`	`576`	`def test_sql(self, mocker):`
`577`	`577`	`# Arrange`