From d930064d503ccf4d64622a57d9bdeecd54692df2 Mon Sep 17 00:00:00 2001 From: Achal Shah Date: Mon, 6 Jun 2022 11:34:06 -0700 Subject: [PATCH] tests Signed-off-by: Achal Shah --- .../tests/integration/registration/test_sql_registry.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/sdk/python/tests/integration/registration/test_sql_registry.py b/sdk/python/tests/integration/registration/test_sql_registry.py index efad9f2c81..029efff596 100644 --- a/sdk/python/tests/integration/registration/test_sql_registry.py +++ b/sdk/python/tests/integration/registration/test_sql_registry.py @@ -262,6 +262,12 @@ def location_features_from_push(inputs: pd.DataFrame) -> pd.DataFrame: # Register Feature View sql_registry.apply_feature_view(location_features_from_push, project) + assert not sql_registry.get_user_metadata(project, location_features_from_push) + + b = "metadata".encode("utf-8") + sql_registry.apply_user_metadata(project, location_features_from_push, b) + assert sql_registry.get_user_metadata(project, location_features_from_push) == b + feature_views = sql_registry.list_on_demand_feature_views(project) # List Feature Views