From d8ddf1fcb0a516269c8a041ba2ca1a36931af87e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=88=98=E6=A3=92=E6=A3=92?= <liubangbang@fuzhi.ai>
Date: Wed, 22 Nov 2023 20:35:47 +0800
Subject: [PATCH] add new test for list plan.

---
 .../actions/test_write_code_function.py       | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/tests/metagpt/actions/test_write_code_function.py b/tests/metagpt/actions/test_write_code_function.py
index 0e57b4ced..cac459380 100644
--- a/tests/metagpt/actions/test_write_code_function.py
+++ b/tests/metagpt/actions/test_write_code_function.py
@@ -1,6 +1,7 @@
 import pytest
 
 from metagpt.actions.write_code_function import WriteCodeFunction
+from metagpt.actions.code_executor import PyCodeExecutor
 
 
 @pytest.mark.asyncio
@@ -20,3 +21,21 @@ async def test_write_code_by_list_prompt():
     assert "language" in code.content
     assert "code" in code.content
     print(code)
+
+
+@pytest.mark.asyncio
+async def test_write_code_by_list_plan():
+    coder = WriteCodeFunction()
+    executor = PyCodeExecutor()
+    messages = []
+    plan = ["随机生成一个pandas DataFrame时间序列", "绘制这个时间序列的直方图", "求均值"]
+    for task in plan:
+        print(f"\n任务: {task}\n\n")
+        messages.append(task)
+        code = await coder.run(messages)
+        messages.append(code)
+        assert "language" in code.content
+        assert "code" in code.content
+        output = await executor.run(code)
+        print(f"\n[Output]: 任务{task}的执行结果是: \n{output}\n")
+        messages.append(output)