apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-15 11:02:36 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix tests

Browse source
This commit is contained in:
geekan 2023-12-29 10:56:44 +08:00
parent 77c537d202
commit e399163d3a
2 changed files with 4 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

3
tests/metagpt/actions/test_write_code_review.py
View file

@ -8,8 +8,7 @@
import pytest
from metagpt.actions.write_code_review import WriteCodeReview
from metagpt.document import Document
from metagpt.schema import CodingContext
from metagpt.schema import CodingContext, Document
@pytest.mark.asyncio

14
tests/metagpt/roles/test_engineer.py
View file

@ -12,12 +12,7 @@ import pytest
from metagpt.logs import logger
from metagpt.roles.engineer import Engineer
from metagpt.utils.common import CodeParser
from tests.metagpt.roles.mock import (
STRS_FOR_PARSING,
TASKS,
TASKS_TOMATO_CLOCK,
MockMessages,
)
from tests.metagpt.roles.mock import STRS_FOR_PARSING, TASKS, MockMessages
@pytest.mark.asyncio
@ -62,14 +57,11 @@ target_list = [
def test_parse_file_list():
tasks = CodeParser.parse_file_list("任务列表", TASKS)
tasks = CodeParser.parse_file_list("Task list", TASKS)
logger.info(tasks)
assert isinstance(tasks, list)
assert target_list == tasks
file_list = CodeParser.parse_file_list("Task list", TASKS_TOMATO_CLOCK, lang="python")
logger.info(file_list)
target_code = """task_list = [
"smart_search_engine/knowledge_base.py",
@ -88,7 +80,7 @@ target_code = """task_list = [
def test_parse_code():
code = CodeParser.parse_code("任务列表", TASKS, lang="python")
code = CodeParser.parse_code("Task list", TASKS, lang="python")
logger.info(code)
assert isinstance(code, str)
assert target_code == code