From 402ec5bcb44528f9c2ce7505e75f30998cd87024 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=88=98=E6=A3=92=E6=A3=92?= Date: Mon, 11 Dec 2023 14:39:57 +0800 Subject: [PATCH] add new test for make tools. --- tests/metagpt/actions/test_make_tools.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/tests/metagpt/actions/test_make_tools.py b/tests/metagpt/actions/test_make_tools.py index 2c5168bf1..4f7d7859a 100644 --- a/tests/metagpt/actions/test_make_tools.py +++ b/tests/metagpt/actions/test_make_tools.py @@ -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)