From 4df1213c86e2cda06938368072fffb8a9219c7ee Mon Sep 17 00:00:00 2001
From: geekan <geekan@foxmail.com>
Date: Tue, 9 Jan 2024 17:56:58 +0800
Subject: [PATCH] refine code

---
 tests/metagpt/test_config.py | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/tests/metagpt/test_config.py b/tests/metagpt/test_config.py
index 85e32818d..5492d1726 100644
--- a/tests/metagpt/test_config.py
+++ b/tests/metagpt/test_config.py
@@ -34,7 +34,7 @@ class NewModel(ConfigMixin, BaseModel):
     b: str = "b"
 
 
-def test_config_mixin():
+def test_config_mixin_1():
     new_model = NewModel()
     assert new_model.a == "a"
     assert new_model.b == "b"
@@ -44,7 +44,12 @@ def test_config_mixin():
 
 def test_config_mixin_2():
     i = Config(llm={"default": mock_llm_config})
-    new_model = NewModel(config=i)
-    assert new_model.config == i
-    assert new_model._config == i
-    assert new_model.config.llm["default"] == mock_llm_config
+    j = Config(llm={"new": mock_llm_config})
+    obj = NewModel(config=i)
+    assert obj.config == i
+    assert obj._config == i
+    assert obj.config.llm["default"] == mock_llm_config
+
+    obj.try_set_parent_config(j)
+    # obj already has a config, so it will not be set
+    assert obj.config == i