diff --git a/feathr_project/test/test_registry_client.py b/feathr_project/test/test_registry_client.py index 2944bac42..3fcc08aca 100644 --- a/feathr_project/test/test_registry_client.py +++ b/feathr_project/test/test_registry_client.py @@ -10,6 +10,7 @@ from feathr.definition.source import INPUT_CONTEXT, HdfsSource, InputContext from feathr.definition.transformation import ExpressionTransformation, WindowAggTransformation from feathr.definition.typed_key import TypedKey +import unittest, pytest from feathr.registry._feathr_registry_client import ( _FeatureRegistry, @@ -219,7 +220,7 @@ def test_parse_project(): ) assert len(derived_features) == 3 - +@pytest.mark.skip(reason="Skip since we cannot connect to external database") def test_registry_client_list_features(): c = _FeatureRegistry(project_name="p", endpoint="https://feathr-sql-registry.azurewebsites.net/api/v1") f = [e["qualifiedName"] for e in c.list_registered_features("feathr_ci_registry_getting_started")] @@ -228,7 +229,7 @@ def test_registry_client_list_features(): for i in f: assert i.startswith("feathr_ci_registry_getting_started__") - +@pytest.mark.skip(reason="Skip since we cannot connect to external database") def test_registry_client_load(): c = _FeatureRegistry(project_name="p", endpoint="https://feathr-sql-registry.azurewebsites.net/api/v1") (anchors, derived_features) = c.get_features_from_registry("feathr_ci_registry_getting_started") @@ -246,7 +247,7 @@ def test_registry_client_load(): ) assert len(derived_features) == 2 - +@pytest.mark.skip(reason="Skip since we cannot connect to external database") def test_create(): project_name = f"feathr_registry_client_test_{int(time.time())}" c = _FeatureRegistry(project_name="p", endpoint="https://feathr-sql-registry.azurewebsites.net/api/v1") diff --git a/registry/test/test_sql_registry.py b/registry/test/test_sql_registry.py index 58e0b2a70..e0d56773e 100644 --- a/registry/test/test_sql_registry.py +++ b/registry/test/test_sql_registry.py @@ -68,7 +68,7 @@ def create_and_get_derived_feature(self, project_id, qualified_name, name, featu af1_downstream_entities = self.registry.get_dependent_entities(af_id) assert len(af1_downstream_entities) > 0 return df_id - + @pytest.mark.skip(reason="Skip since we cannot connect to external database") def test_registry(self): self.setup() now = datetime.now()