Unset project as cleanup in empty_data_context

great-expectations · May 9, 2024 · 41b1ebe · 41b1ebe
1 parent cf42fc0
commit 41b1ebe
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/great_expectations/data_context/data_context/context_factory.py b/great_expectations/data_context/data_context/context_factory.py
@@ -82,7 +82,7 @@ def get_project(  # noqa: PLR0913
         )
         return self.__project
 
-    def set_project(self, project: AbstractDataContext) -> None:
+    def set_project(self, project: AbstractDataContext | None) -> None:
         self.__project = project
 
     @property

diff --git a/tests/conftest.py b/tests/conftest.py
@@ -723,7 +723,7 @@ def mysql_engine(test_backend):
 @pytest.fixture(scope="function")
 def empty_data_context(
     tmp_path,
-) -> FileDataContext:
+) -> Generator[FileDataContext, None, None]:
     project_path = tmp_path / "empty_data_context"
     project_path.mkdir()
     project_path = str(project_path)
@@ -733,7 +733,8 @@ def empty_data_context(
     os.makedirs(asset_config_path, exist_ok=True)  # noqa: PTH103
     assert context.list_datasources() == []
     project_manager.set_project(context)
-    return context
+    yield context
+    project_manager.set_project(None)
 
 
 @pytest.fixture(scope="function")