apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-11 15:15:18 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

add new test for make tools.

Browse source
This commit is contained in:
刘棒棒 2023-12-11 14:39:57 +08:00
parent ab020adec4
commit 402ec5bcb4
1 changed files with 16 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

16
tests/metagpt/actions/test_make_tools.py
View file

@ -16,3 +16,19 @@ async def test_make_tools():
result, res_type = await ep.run(tool_code)
assert res_type is True
print(result)
@pytest.mark.asyncio
async def test_make_tools2():
code = '''import pandas as pd\npath = "./tests/data/test.csv"\ndf = pd.read_csv(path)\ndata = df.copy()\n
data['started_at'] = data['started_at'].apply(lambda r: pd.to_datetime(r))\n
data['ended_at'] = data['ended_at'].apply(lambda r: pd.to_datetime(r))\ndata.head()'''
msgs = [{'role': 'assistant', 'content': code}]
mt = MakeTools()
tool_code = await mt.run(msgs)
print(tool_code)
ep = ExecutePyCode()
tool_code = tool_code
result, res_type = await ep.run(tool_code)
assert res_type is True
print(result)