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

refine parse_outputs in ExecuteNbCode.

Browse source
This commit is contained in:
刘棒棒 2024-03-11 13:57:25 +08:00
parent 543f519deb
commit bcde5171e0
2 changed files with 33 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

17
tests/metagpt/actions/mi/test_execute_nb_code.py
View file

@ -100,6 +100,7 @@ async def test_terminate():
is_kernel_alive = await executor.nb_client.km.is_alive()
assert is_kernel_alive
await executor.terminate()
import time
time.sleep(2)
@ -123,3 +124,19 @@ async def test_reset():
assert is_kernel_alive
await executor.reset()
assert executor.nb_client.km is None
@pytest.mark.asyncio
async def test_parse_outputs():
executor = ExecuteNbCode()
code = """
import pandas as pd
df = pd.DataFrame({'ID': [1,2,3], 'NAME': ['a', 'b', 'c']})
print(df.columns)
print(df['DUMMPY_ID'])
"""
output, is_success = await executor.run(code)
assert not is_success
assert "Index(['ID', 'NAME'], dtype='object')" in output
assert "Executed code failed," in output
assert "KeyError: 'DUMMPY_ID'" in output