refine code

tests/metagpt/test_config.py

@@ -5,8 +5,9 @@
 @Author  : alexanderwu
 @File    : test_config.py
 """
+from pydantic import BaseModel
 
-from metagpt.config2 import Config, config
+from metagpt.config2 import Config, ConfigMixin, config
 from metagpt.configs.llm_config import LLMType
 from tests.metagpt.provider.mock_llm_config import mock_llm_config
 
@@ -26,3 +27,24 @@ def test_config_from_dict():
     cfg = Config(llm={"default": mock_llm_config})
     assert cfg
     assert cfg.llm["default"].api_key == "mock_api_key"
+
+
+class NewModel(ConfigMixin, BaseModel):
+    a: str = "a"
+    b: str = "b"
+
+
+def test_config_mixin():
+    new_model = NewModel()
+    assert new_model.a == "a"
+    assert new_model.b == "b"
+    assert new_model._config == new_model.config
+    assert new_model._config is None
+
+
+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