This commit is contained in:
femto 2023-09-08 17:21:42 +08:00
parent dd8be8a382
commit 2846a462a0
4 changed files with 24 additions and 12 deletions

View file

@ -7,6 +7,7 @@
"""
import pytest
from metagpt.actions import ActionOutput
from metagpt.actions.design_api import WriteDesign
from metagpt.logs import logger
from tests.metagpt.actions.mock import PRD_SAMPLE
@ -18,9 +19,10 @@ async def test_design_api():
design_api = WriteDesign("design_api")
result = await design_api.run(prd)
result = await design_api.run([ActionOutput(content=prd, instruct_content=None)])
logger.info(result)
assert len(result) > 0
assert result
@pytest.mark.asyncio
@ -28,7 +30,7 @@ async def test_design_api_calculator():
prd = PRD_SAMPLE
design_api = WriteDesign("design_api")
result = await design_api.run(prd)
result = await design_api.run([ActionOutput(content=prd, instruct_content=None)])
logger.info(result)
assert len(result) > 10
assert result