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

pre-commit format

Browse source
This commit is contained in:
yzlin 2024-01-13 10:37:15 +08:00
parent e079b8b152
commit 4b09ffd2f5
2 changed files with 9 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

15
tests/metagpt/actions/test_debug_code.py
View file

@ -8,13 +8,13 @@ import pytest
from metagpt.actions.debug_code import DebugCode, messages_to_str
from metagpt.schema import Message
ErrorStr = '''Tested passed:
ErrorStr = """Tested passed:
Tests failed:
assert sort_array([1, 5, 2, 3, 4]) == [1, 2, 3, 4, 5] # output: [1, 2, 4, 3, 5]
'''
"""
CODE = '''
CODE = """
def sort_array(arr):
# Helper function to count the number of ones in the binary representation
def count_ones(n):
@ -27,7 +27,7 @@ def sort_array(arr):
return sorted_arr
```
'''
"""
DebugContext = '''Solve the problem in Python:
def sort_array(arr):
@ -42,13 +42,16 @@ def sort_array(arr):
>>> sort_array([1, 0, 2, 3, 4]) [0, 1, 2, 3, 4]
"""
'''
@pytest.mark.asyncio
async def test_debug_code():
debug_context = Message(content=DebugContext)
new_code = await DebugCode().run(context=debug_context, code=CODE, runtime_result=ErrorStr)
assert "def sort_array(arr)" in new_code
def test_messages_to_str():
debug_context = Message(content=DebugContext)
msg_str = messages_to_str([debug_context])
assert "user: Solve the problem in Python" in msg_str
assert "user: Solve the problem in Python" in msg_str