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

save code

Browse source
This commit is contained in:
程茂宇 2023-07-24 14:21:18 +08:00
parent 8b876ac6b9
commit b187f80f30
4 changed files with 56 additions and 43 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -13,23 +13,29 @@ from tests.metagpt.actions.mock import SEARCH_CODE_SAMPLE
@pytest.mark.asyncio
async def test_write_code_review():
async def test_write_code_review(capfd):
code = """
def add(a, b):
return a + b
return a +
"""
write_code_review = WriteCodeReview("write_code_review")
# write_code_review = WriteCodeReview("write_code_review")
review = await write_code_review.run(code)
code = await WriteCodeReview().run(
context="编写一个从a加b的函数，返回a+b",
code=code,
filename="math.py"
)
# 我们不能精确地预测生成的代码评审，但我们可以检查返回的是否为字符串
assert isinstance(review, str)
assert len(review) > 0
assert isinstance(code, str)
assert len(code) > 0
captured = capfd.readouterr()
print(f"输出内容: {captured.out}")
@pytest.mark.asyncio
async def test_write_code_review_directly():
code = SEARCH_CODE_SAMPLE
write_code_review = WriteCodeReview("write_code_review")
review = await write_code_review.run(code)
logger.info(review)
# @pytest.mark.asyncio
# async def test_write_code_review_directly():
# code = SEARCH_CODE_SAMPLE
# write_code_review = WriteCodeReview("write_code_review")
# review = await write_code_review.run(code)
# logger.info(review)