From 5af4f6b4c524e62dd43ff6f6f6e80062f8427ead 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, 20 Dec 2023 19:56:26 +0800
Subject: [PATCH] add new test for aask_code about write code by steps.

---
 tests/metagpt/provider/test_openai.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/tests/metagpt/provider/test_openai.py b/tests/metagpt/provider/test_openai.py
index 2b0af37b5..98a3670f1 100644
--- a/tests/metagpt/provider/test_openai.py
+++ b/tests/metagpt/provider/test_openai.py
@@ -78,3 +78,17 @@ def test_ask_code_list_str():
     assert "language" in rsp
     assert "code" in rsp
     assert len(rsp["code"]) > 0
+
+
+@pytest.mark.asyncio
+async def test_ask_code_steps2():
+    llm = OpenAIGPTAPI()
+    msg = ["step by setp 生成代码: Step 1. 先生成随机数组a, Step 2. 求a中最大值, Step 3. 绘制数据a的直方图"]
+    rsp = await llm.aask_code(msg)  # -> {'language': 'python', 'code': 'max_value = max(a)\nmax_value'}
+    print(rsp)
+    assert "language" in rsp
+    assert "code" in rsp
+    assert len(rsp["code"]) > 0
+    assert "Step 1" in rsp["code"]
+    assert "Step 2" in rsp["code"]
+    assert "Step 3" in rsp["code"]