diff --git a/tests/metagpt/roles/test_qa_engineer.py b/tests/metagpt/roles/test_qa_engineer.py index 02bf37b85..0501ce5a0 100644 --- a/tests/metagpt/roles/test_qa_engineer.py +++ b/tests/metagpt/roles/test_qa_engineer.py @@ -7,20 +7,16 @@ """ from pathlib import Path from typing import List -import json import pytest from pydantic import BaseModel, Field -from unittest.mock import AsyncMock, MagicMock from metagpt.actions import DebugError, RunCode, WriteTest, UserRequirement from metagpt.actions.summarize_code import SummarizeCode -from metagpt.actions.prepare_documents import PrepareDocuments from metagpt.environment import Environment from metagpt.roles import QaEngineer -from metagpt.schema import Message, AIMessage, Document, RunCodeContext, TestingContext -from metagpt.utils.common import any_to_str, aread, awrite -from metagpt.utils.project_repo import ProjectRepo +from metagpt.schema import Message, AIMessage, Document, RunCodeContext +from metagpt.utils.common import any_to_str, awrite class MockProjectRepo: @@ -220,4 +216,4 @@ async def test_qa_full_workflow(tmp_path): env.msgs.clear() role.test_round_allowed = 1 rsp = await role.run(with_message=msg) - assert "Exceeding" in rsp.content \ No newline at end of file + assert "Exceeding" in rsp.content