format serialize_deserialize tests code

This commit is contained in:
better629 2023-12-19 14:04:09 +08:00
parent b4322bca54
commit 35ac28c30e
9 changed files with 21 additions and 28 deletions

View file

@ -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():

View file

@ -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

View file

@ -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

View file

@ -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

View file

@ -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():

View file

@ -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

View file

@ -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

View file

@ -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"

View file

@ -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