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

rm unused & format

Browse source
This commit is contained in:
yzlin 2024-01-30 22:41:30 +08:00
parent 274747e72f
commit 5dde5a8875
2 changed files with 2 additions and 54 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1,52 +0,0 @@
import pytest
from metagpt.actions.execute_code import ExecutePyCode
from metagpt.actions.write_analysis_code import MakeTools
from metagpt.logs import logger
@pytest.mark.asyncio
async def test_make_tools():
code = "import yfinance as yf\n\n# Collect Alibaba stock data\nalibaba = yf.Ticker('BABA')\ndata = alibaba.history(period='1d', start='2022-01-01', end='2022-12-31')\nprint(data.head())"
msgs = [{"role": "assistant", "content": code}]
mt = MakeTools()
tool_code = await mt.run(msgs)
logger.debug(tool_code)
ep = ExecutePyCode()
tool_code = "!pip install yfinance\n" + tool_code
result, res_type = await ep.run(tool_code)
assert res_type is True
logger.debug(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)
logger.debug(tool_code)
ep = ExecutePyCode()
tool_code = tool_code
result, res_type = await ep.run(tool_code)
assert res_type is True
logger.debug(result)
@pytest.mark.asyncio
async def test_make_tools3():
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))\n
data['duration_hour'] = (data['ended_at'] - data['started_at']).dt.seconds/3600\ndata.head()"""
msgs = [{"role": "assistant", "content": code}]
mt = MakeTools()
tool_code = await mt.run(msgs)
logger.debug(tool_code)
ep = ExecutePyCode()
tool_code = tool_code
result, res_type = await ep.run(tool_code)
assert res_type is True
logger.debug(result)