ishangsf/workflows
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Skip 2 llm calls temporarily

Browse Source
This commit is contained in:
kagami 2024-05-19 16:50:30 +08:00
parent da12ac1fb9
commit 2d97bc40e6
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
merico/unit_tests/ut_workflow.py
View File

@ -6,7 +6,6 @@ from find_context import (
Position, Position,
Range, Range,
find_symbol_context_by_static_analysis, find_symbol_context_by_static_analysis,
find_symbol_context_of_llm_recommendation,
) )
from find_reference_tests import find_reference_tests from find_reference_tests import find_reference_tests
from i18n import TUILanguage, get_translation from i18n import TUILanguage, get_translation
@ -52,8 +51,10 @@ class UnitTestsWorkflow:
contexts.update(v) contexts.update(v)
contexts = list(contexts) contexts = list(contexts)
print("- Finding reference files...", flush=True) reference_files = []
reference_files = self.step_2_find_reference_files() # TODO: Restore this feature after checking the performance issue
# print("- Finding reference files...", flush=True)
# reference_files = self.step_2_find_reference_files()
print("- Proposing test cases...", flush=True) print("- Proposing test cases...", flush=True)
cases = self.step_3_propose_cases(contexts) cases = self.step_3_propose_cases(contexts)
@ -88,11 +89,12 @@ class UnitTestsWorkflow:
{item for sublist in list(symbol_context.values()) for item in sublist} {item for sublist in list(symbol_context.values()) for item in sublist}
) )
recommended_context = find_symbol_context_of_llm_recommendation( # TODO: Restore this feature after checking the performance issue
self.func_to_test, known_context_for_llm # recommended_context = find_symbol_context_of_llm_recommendation(
) # self.func_to_test, known_context_for_llm
# )
symbol_context.update(recommended_context) # symbol_context.update(recommended_context)
return symbol_context return symbol_context