From d8abbd30636c71331768c8f78a36240c1b39b5e5 Mon Sep 17 00:00:00 2001 From: Huong Nguyen Date: Tue, 3 Dec 2024 10:28:55 +0000 Subject: [PATCH] update test_server Signed-off-by: Huong Nguyen --- package/tests/test_server.py | 20 +------------------- 1 file changed, 1 insertion(+), 19 deletions(-) diff --git a/package/tests/test_server.py b/package/tests/test_server.py index 011b05def..1c0960c1c 100644 --- a/package/tests/test_server.py +++ b/package/tests/test_server.py @@ -31,30 +31,12 @@ def patched_create_api_app_from_file(mocker): @pytest.fixture(autouse=True) -def patched_load_data( - mocker, example_catalog, example_pipelines, session_store, example_stats_dict -): +def patched_load_data(mocker, example_catalog, example_pipelines, example_stats_dict): yield mocker.patch( "kedro_viz.server.kedro_data_loader.load_data", return_value=( example_catalog, example_pipelines, - session_store, - example_stats_dict, - ), - ) - - -@pytest.fixture -def patched_load_data_with_sqlite_session_store( - mocker, example_catalog, example_pipelines, sqlite_session_store, example_stats_dict -): - yield mocker.patch( - "kedro_viz.server.kedro_data_loader.load_data", - return_value=( - example_catalog, - example_pipelines, - sqlite_session_store, example_stats_dict, ), )