Merge pull request #73 from qa6300525/2023-07-10_chengmaoyu

2023 07 10 chengmaoyu
This commit is contained in:
geekan 2023-07-25 11:03:44 +08:00 committed by GitHub
commit a538f9a3ca
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
7 changed files with 175 additions and 68 deletions

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)