From 9b2c8db70d5089c2c1b7be43cc92ea1b0c6f1fab Mon Sep 17 00:00:00 2001 From: David Y Liu Date: Fri, 25 Jun 2021 10:43:45 -0700 Subject: [PATCH] update in testing Signed-off-by: David Y Liu --- sdk/python/tests/test_inference.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/sdk/python/tests/test_inference.py b/sdk/python/tests/test_inference.py index 507283b2af..9405bce1f2 100644 --- a/sdk/python/tests/test_inference.py +++ b/sdk/python/tests/test_inference.py @@ -14,7 +14,9 @@ ) -def test_infer_entity_value_type_from_feature_views(simple_dataset_1, simple_dataset_2): +def test_update_entities_with_inferred_types_from_feature_views( + simple_dataset_1, simple_dataset_2 +): with prep_file_source( df=simple_dataset_1, event_timestamp_column="ts_1" ) as file_source, prep_file_source( @@ -29,8 +31,8 @@ def test_infer_entity_value_type_from_feature_views(simple_dataset_1, simple_dat update_entities_with_inferred_types_from_feature_views([actual_1], [fv1]) update_entities_with_inferred_types_from_feature_views([actual_2], [fv2]) - assert actual_1 == [Entity(name="id", value_type=ValueType.INT64)] - assert actual_2 == [Entity(name="id", value_type=ValueType.STRING)] + assert actual_1 == Entity(name="id", value_type=ValueType.INT64) + assert actual_2 == Entity(name="id", value_type=ValueType.STRING) with pytest.raises(RegistryInferenceFailure): # two viable data types @@ -40,7 +42,7 @@ def test_infer_entity_value_type_from_feature_views(simple_dataset_1, simple_dat @pytest.mark.integration -def test_infer_event_timestamp_column_for_data_source(simple_dataset_1): +def test_update_data_sources_with_inferred_event_timestamp_col(simple_dataset_1): df_with_two_viable_timestamp_cols = simple_dataset_1.copy(deep=True) df_with_two_viable_timestamp_cols["ts_2"] = simple_dataset_1["ts_1"]