From e399163d3a40285702f03df4131cae042fdb104b Mon Sep 17 00:00:00 2001
From: geekan <geekan@foxmail.com>
Date: Fri, 29 Dec 2023 10:56:44 +0800
Subject: [PATCH] fix tests

---
 tests/metagpt/actions/test_write_code_review.py |  3 +--
 tests/metagpt/roles/test_engineer.py            | 14 +++-----------
 2 files changed, 4 insertions(+), 13 deletions(-)

diff --git a/tests/metagpt/actions/test_write_code_review.py b/tests/metagpt/actions/test_write_code_review.py
index e16eb7348..3343b42b4 100644
--- a/tests/metagpt/actions/test_write_code_review.py
+++ b/tests/metagpt/actions/test_write_code_review.py
@@ -8,8 +8,7 @@
 import pytest
 
 from metagpt.actions.write_code_review import WriteCodeReview
-from metagpt.document import Document
-from metagpt.schema import CodingContext
+from metagpt.schema import CodingContext, Document
 
 
 @pytest.mark.asyncio
diff --git a/tests/metagpt/roles/test_engineer.py b/tests/metagpt/roles/test_engineer.py
index 3dc599770..6e7bc49ea 100644
--- a/tests/metagpt/roles/test_engineer.py
+++ b/tests/metagpt/roles/test_engineer.py
@@ -12,12 +12,7 @@ import pytest
 from metagpt.logs import logger
 from metagpt.roles.engineer import Engineer
 from metagpt.utils.common import CodeParser
-from tests.metagpt.roles.mock import (
-    STRS_FOR_PARSING,
-    TASKS,
-    TASKS_TOMATO_CLOCK,
-    MockMessages,
-)
+from tests.metagpt.roles.mock import STRS_FOR_PARSING, TASKS, MockMessages
 
 
 @pytest.mark.asyncio
@@ -62,14 +57,11 @@ target_list = [
 
 
 def test_parse_file_list():
-    tasks = CodeParser.parse_file_list("任务列表", TASKS)
+    tasks = CodeParser.parse_file_list("Task list", TASKS)
     logger.info(tasks)
     assert isinstance(tasks, list)
     assert target_list == tasks
 
-    file_list = CodeParser.parse_file_list("Task list", TASKS_TOMATO_CLOCK, lang="python")
-    logger.info(file_list)
-
 
 target_code = """task_list = [
     "smart_search_engine/knowledge_base.py",
@@ -88,7 +80,7 @@ target_code = """task_list = [
 
 
 def test_parse_code():
-    code = CodeParser.parse_code("任务列表", TASKS, lang="python")
+    code = CodeParser.parse_code("Task list", TASKS, lang="python")
     logger.info(code)
     assert isinstance(code, str)
     assert target_code == code