Improve writing test prompt

merico/unit_tests/main.py

@@ -7,7 +7,6 @@
 
 sys.path.append(os.path.dirname(__file__))
 
-from cache import LocalCache
 from i18n import TUILanguage, get_translation
 from model import (
     FuncToTest,
@@ -16,6 +15,7 @@
 )
 from ut_workflow import UnitTestsWorkflow
 
+from cache import LocalCache
 from lib.chatmark import Step
 from lib.ide_service import IDEService
 

merico/unit_tests/prompts.py

@@ -76,6 +76,7 @@
 5. If two or more test cases share the same setup, you should reuse the setup code.
 6. If two or more test cases share the same test logic, you should reuse the test logic.
 7. Use TODO comments or FIXME comments to indicate any missing parts of the code or any parts that need to be improved.
+8. Write as much production-ready code as possible, leaving less TODO comments or FIXME comments.
 
 {additional_requirements}
 

merico/unit_tests/ut_workflow.py

@@ -1,6 +1,5 @@
 from typing import Dict, List, Tuple
 
-from cache import LocalCache
 from find_context import (
     Context,
     Position,
@@ -18,6 +17,7 @@
 from tools.file_util import retrieve_file_content
 from write_tests import write_and_print_tests
 
+from cache import LocalCache
 from lib.chatmark import Checkbox, Form, Step, TextEditor