chore: Updated unit test (feast-dev#4586)

franciscojavierarceo · web-flow · commit 2ef8dac546a7 · 2024-09-29T15:06:07.000-04:00
updated unit test

Signed-off-by: Francisco Javier Arceo &lt;farceo@redhat.com&gt;
diff --git a/sdk/python/tests/unit/online_store/test_online_writes.py b/sdk/python/tests/unit/online_store/test_online_writes.py
@@ -75,6 +75,8 @@ def setUp(self):
                 online=True,
                 source=driver_stats_source,
             )
+            # Before apply() join_keys is empty
+            assert driver_stats_fv.join_keys == []
 
             @on_demand_feature_view(
                 sources=[driver_stats_fv[["conv_rate", "acc_rate"]]],
@@ -100,6 +102,8 @@ def test_view(inputs: dict[str, Any]) -> dict[str, Any]:
                     test_view,
                 ]
             )
+            # after apply() join_keys is [driver]
+            assert driver_stats_fv.join_keys == [driver]
             self.store.write_to_online_store(
                 feature_view_name="driver_hourly_stats", df=driver_df
             )

Original file line number	Diff line number	Diff line change
`@@ -75,6 +75,8 @@ def setUp(self):`
`75`	`75`	`online=True,`
`76`	`76`	`source=driver_stats_source,`
`77`	`77`	`)`
	`78`	`+ # Before apply() join_keys is empty`
	`79`	`+ assert driver_stats_fv.join_keys == []`
`78`	`80`
`79`	`81`	`@on_demand_feature_view(`
`80`	`82`	`sources=[driver_stats_fv[["conv_rate", "acc_rate"]]],`
`@@ -100,6 +102,8 @@ def test_view(inputs: dict[str, Any]) -> dict[str, Any]:`
`100`	`102`	`test_view,`
`101`	`103`	`]`
`102`	`104`	`)`
	`105`	`+ # after apply() join_keys is [driver]`
	`106`	`+ assert driver_stats_fv.join_keys == [driver]`
`103`	`107`	`self.store.write_to_online_store(`
`104`	`108`	`feature_view_name="driver_hourly_stats", df=driver_df`
`105`	`109`	`)`