updated local feature store test

franciscojavierarceo · franciscojavierarceo · commit acc7b7f7f1ce · 2024-10-05T05:41:18.000-04:00
Signed-off-by: Francisco Javier Arceo &lt;farceo@redhat.com&gt;
diff --git a/sdk/python/tests/unit/local_feast_tests/test_local_feature_store.py b/sdk/python/tests/unit/local_feast_tests/test_local_feature_store.py
@@ -386,7 +386,7 @@ def test_apply_dummy_entity_and_feature_view_columns(test_feature_store):
     assert fv_with_entity.entity_columns == []
 
     # Register Feature View
-    test_feature_store.apply([e1, fv_no_entity])
+    test_feature_store.apply([e1, fv_no_entity, fv_with_entity])
     fv_from_online_store = test_feature_store.get_feature_view(
         "my_feature_view_no_entity"
     )
@@ -395,6 +395,7 @@ def test_apply_dummy_entity_and_feature_view_columns(test_feature_store):
     assert fv_from_online_store.entity_columns[0].name == DUMMY_ENTITY_ID
     assert fv_from_online_store.entities == []
     assert fv_no_entity.entities == [DUMMY_ENTITY_NAME]
+    assert fv_with_entity.entity_columns[0] == e1.join_key
 
     test_feature_store.teardown()
 
diff --git a/sdk/python/tests/unit/test_on_demand_python_transformation.py b/sdk/python/tests/unit/test_on_demand_python_transformation.py
@@ -29,8 +29,8 @@
     Int64,
     String,
     UnixTimestamp,
-    _utc_now,
     ValueType,
+    _utc_now,
     from_value_type,
 )
 
