From 41b1ebea1bae9673771cc04d8df1093477dac71a Mon Sep 17 00:00:00 2001 From: Tyler Hoffman Date: Thu, 9 May 2024 11:03:31 -0400 Subject: [PATCH] Unset project as cleanup in empty_data_context --- .../data_context/data_context/context_factory.py | 2 +- tests/conftest.py | 5 +++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/great_expectations/data_context/data_context/context_factory.py b/great_expectations/data_context/data_context/context_factory.py index 7896410bb24e..ba00278bb2df 100644 --- 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 index 6315b9f3f2a5..cd557918746a 100644 --- 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")