diff --git a/tests/metagpt/serialize_deserialize/test_architect_deserialize.py b/tests/metagpt/serialize_deserialize/test_architect_deserialize.py index 66fba6167..b92eba8a1 100644 --- a/tests/metagpt/serialize_deserialize/test_architect_deserialize.py +++ b/tests/metagpt/serialize_deserialize/test_architect_deserialize.py @@ -4,8 +4,8 @@ # @Desc : import pytest -from metagpt.roles.architect import Architect from metagpt.actions.action import Action +from metagpt.roles.architect import Architect def test_architect_serialize(): diff --git a/tests/metagpt/serialize_deserialize/test_environment.py b/tests/metagpt/serialize_deserialize/test_environment.py index 4e3445047..3a374460c 100644 --- a/tests/metagpt/serialize_deserialize/test_environment.py +++ b/tests/metagpt/serialize_deserialize/test_environment.py @@ -2,17 +2,15 @@ # -*- coding: utf-8 -*- # @Desc : -from pathlib import Path import shutil -from metagpt.schema import Message from metagpt.actions.action_output import ActionOutput -from metagpt.roles.project_manager import ProjectManager from metagpt.actions.add_requirement import UserRequirement from metagpt.actions.project_management import WriteTasks from metagpt.environment import Environment +from metagpt.roles.project_manager import ProjectManager +from metagpt.schema import Message from metagpt.utils.common import any_to_str - from tests.metagpt.serialize_deserialize.test_serdeser_base import RoleC, ActionOK, serdeser_path diff --git a/tests/metagpt/serialize_deserialize/test_memory.py b/tests/metagpt/serialize_deserialize/test_memory.py index 50d30a94d..47410c615 100644 --- a/tests/metagpt/serialize_deserialize/test_memory.py +++ b/tests/metagpt/serialize_deserialize/test_memory.py @@ -2,16 +2,14 @@ # -*- coding: utf-8 -*- # @Desc : unittest of memory -from pathlib import Path from pydantic import BaseModel -from metagpt.schema import Message -from metagpt.memory.memory import Memory from metagpt.actions.action_output import ActionOutput -from metagpt.actions.design_api import WriteDesign from metagpt.actions.add_requirement import UserRequirement +from metagpt.actions.design_api import WriteDesign +from metagpt.memory.memory import Memory +from metagpt.schema import Message from metagpt.utils.common import any_to_str - from tests.metagpt.serialize_deserialize.test_serdeser_base import serdeser_path diff --git a/tests/metagpt/serialize_deserialize/test_product_manager.py b/tests/metagpt/serialize_deserialize/test_product_manager.py index 1d721282f..b65e329d1 100644 --- a/tests/metagpt/serialize_deserialize/test_product_manager.py +++ b/tests/metagpt/serialize_deserialize/test_product_manager.py @@ -4,8 +4,8 @@ # @Desc : import pytest -from metagpt.roles.product_manager import ProductManager from metagpt.actions.action import Action +from metagpt.roles.product_manager import ProductManager from metagpt.schema import Message diff --git a/tests/metagpt/serialize_deserialize/test_project_manager.py b/tests/metagpt/serialize_deserialize/test_project_manager.py index 21fafa72e..e52e3f247 100644 --- a/tests/metagpt/serialize_deserialize/test_project_manager.py +++ b/tests/metagpt/serialize_deserialize/test_project_manager.py @@ -4,9 +4,9 @@ # @Desc : import pytest -from metagpt.roles.project_manager import ProjectManager from metagpt.actions.action import Action from metagpt.actions.project_management import WriteTasks +from metagpt.roles.project_manager import ProjectManager def test_project_manager_serialize(): diff --git a/tests/metagpt/serialize_deserialize/test_role.py b/tests/metagpt/serialize_deserialize/test_role.py index fe7b63ef3..f25403dc0 100644 --- a/tests/metagpt/serialize_deserialize/test_role.py +++ b/tests/metagpt/serialize_deserialize/test_role.py @@ -4,18 +4,18 @@ # @Desc : import shutil + import pytest -from metagpt.logs import logger -from metagpt.roles.role import Role from metagpt.actions import WriteCode -from metagpt.schema import Message from metagpt.actions.add_requirement import UserRequirement -from metagpt.roles.product_manager import ProductManager from metagpt.const import SERDESER_PATH +from metagpt.logs import logger from metagpt.roles.engineer import Engineer +from metagpt.roles.product_manager import ProductManager +from metagpt.roles.role import Role +from metagpt.schema import Message from metagpt.utils.utils import format_trackback_info - from tests.metagpt.serialize_deserialize.test_serdeser_base import RoleA, RoleB, RoleC, serdeser_path diff --git a/tests/metagpt/serialize_deserialize/test_schema.py b/tests/metagpt/serialize_deserialize/test_schema.py index 97ca4ea0c..02afa762d 100644 --- a/tests/metagpt/serialize_deserialize/test_schema.py +++ b/tests/metagpt/serialize_deserialize/test_schema.py @@ -2,11 +2,10 @@ # -*- coding: utf-8 -*- # @Desc : unittest of schema ser&deser -from metagpt.schema import Message from metagpt.actions.action_output import ActionOutput from metagpt.actions.write_code import WriteCode +from metagpt.schema import Message from metagpt.utils.common import any_to_str - from tests.metagpt.serialize_deserialize.test_serdeser_base import MockMessage diff --git a/tests/metagpt/serialize_deserialize/test_serdeser_base.py b/tests/metagpt/serialize_deserialize/test_serdeser_base.py index 0363c519b..20f708e30 100644 --- a/tests/metagpt/serialize_deserialize/test_serdeser_base.py +++ b/tests/metagpt/serialize_deserialize/test_serdeser_base.py @@ -2,15 +2,15 @@ # -*- coding: utf-8 -*- # @Desc : base test actions / roles used in unittest -from pydantic import BaseModel, Field -from pathlib import Path import asyncio +from pathlib import Path + +from pydantic import BaseModel, Field from metagpt.actions.action import Action -from metagpt.roles.role import Role, RoleReactMode -from metagpt.actions.add_requirement import UserRequirement from metagpt.actions.action_output import ActionOutput - +from metagpt.actions.add_requirement import UserRequirement +from metagpt.roles.role import Role, RoleReactMode serdeser_path = Path(__file__).absolute().parent.joinpath("..", "..", "data", "serdeser_storage") @@ -51,7 +51,6 @@ class ActionRaise(Action): class RoleA(Role): - name: str = Field(default="RoleA") profile: str = Field(default="Role A") goal: str = "RoleA's goal" diff --git a/tests/metagpt/serialize_deserialize/test_team.py b/tests/metagpt/serialize_deserialize/test_team.py index 777f0f381..01e0a6c70 100644 --- a/tests/metagpt/serialize_deserialize/test_team.py +++ b/tests/metagpt/serialize_deserialize/test_team.py @@ -3,14 +3,13 @@ # @Author : stellahong (stellahong@fuzhi.ai) # @Desc : -from pathlib import Path import shutil + import pytest +from metagpt.const import SERDESER_PATH from metagpt.roles import ProjectManager, ProductManager, Architect from metagpt.team import Team -from metagpt.const import SERDESER_PATH - from tests.metagpt.serialize_deserialize.test_serdeser_base import RoleA, RoleB, RoleC, serdeser_path, ActionOK