From c7519462a2fbc8e4925dab7d97a8332fafb66573 Mon Sep 17 00:00:00 2001 From: shenchucheng Date: Thu, 15 Aug 2024 21:01:45 +0800 Subject: [PATCH] fix rag unittest error --- tests/metagpt/exp_pool/test_decorator.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/tests/metagpt/exp_pool/test_decorator.py b/tests/metagpt/exp_pool/test_decorator.py index 0ca4c6ce1..9d104fca4 100644 --- a/tests/metagpt/exp_pool/test_decorator.py +++ b/tests/metagpt/exp_pool/test_decorator.py @@ -155,7 +155,10 @@ class TestExpCache: @pytest.fixture def mock_config(self, mocker): - return mocker.patch("metagpt.exp_pool.decorator.config") + config = Config.default().model_copy(deep=True) + default = mocker.patch("metagpt.config2.Config.default") + default.return_value = config + return config @pytest.mark.asyncio async def test_exp_cache_disabled(self, mock_config, mock_exp_manager): @@ -171,7 +174,9 @@ class TestExpCache: @pytest.mark.asyncio async def test_exp_cache_enabled_no_perfect_exp(self, mock_config, mock_exp_manager, mock_scorer): + mock_config.exp_pool.enabled = True mock_config.exp_pool.enable_read = True + mock_config.exp_pool.enable_write = True mock_exp_manager.query_exps.return_value = [] @exp_cache(manager=mock_exp_manager, scorer=mock_scorer) @@ -185,6 +190,7 @@ class TestExpCache: @pytest.mark.asyncio async def test_exp_cache_enabled_with_perfect_exp(self, mock_config, mock_exp_manager, mock_perfect_judge): + mock_config.exp_pool.enabled = True mock_config.exp_pool.enable_read = True perfect_exp = Experience(req="test", resp="perfect_result") mock_exp_manager.query_exps.return_value = [perfect_exp]