yangdx
commited on
Commit
·
153eff6
1
Parent(s):
a8fc721
Remove error handling tests from "all" mode
Browse files
test_lightrag_ollama_chat.py
CHANGED
@@ -742,7 +742,7 @@ Configuration file (config.json):
|
|
742 |
nargs="+",
|
743 |
choices=list(get_test_cases().keys()) + ["all"],
|
744 |
default=["all"],
|
745 |
-
help="Test cases to run, options: %(choices)s. Use 'all' to run all tests",
|
746 |
)
|
747 |
return parser.parse_args()
|
748 |
|
@@ -766,21 +766,18 @@ if __name__ == "__main__":
|
|
766 |
|
767 |
try:
|
768 |
if "all" in args.tests:
|
769 |
-
# Run all tests
|
770 |
if OutputControl.is_verbose():
|
771 |
print("\n【Chat API Tests】")
|
772 |
run_test(test_non_stream_chat, "Non-streaming Chat Test")
|
773 |
run_test(test_stream_chat, "Streaming Chat Test")
|
774 |
run_test(test_query_modes, "Chat Query Mode Test")
|
775 |
-
|
776 |
-
run_test(test_stream_error_handling, "Chat Streaming Error Test")
|
777 |
-
|
778 |
if OutputControl.is_verbose():
|
779 |
print("\n【Generate API Tests】")
|
780 |
run_test(test_non_stream_generate, "Non-streaming Generate Test")
|
781 |
run_test(test_stream_generate, "Streaming Generate Test")
|
782 |
run_test(test_generate_with_system, "Generate with System Prompt Test")
|
783 |
-
run_test(test_generate_error_handling, "Generate Error Handling Test")
|
784 |
run_test(test_generate_concurrent, "Generate Concurrent Test")
|
785 |
else:
|
786 |
# Run specified tests
|
|
|
742 |
nargs="+",
|
743 |
choices=list(get_test_cases().keys()) + ["all"],
|
744 |
default=["all"],
|
745 |
+
help="Test cases to run, options: %(choices)s. Use 'all' to run all tests (except error tests)",
|
746 |
)
|
747 |
return parser.parse_args()
|
748 |
|
|
|
766 |
|
767 |
try:
|
768 |
if "all" in args.tests:
|
769 |
+
# Run all tests except error handling tests
|
770 |
if OutputControl.is_verbose():
|
771 |
print("\n【Chat API Tests】")
|
772 |
run_test(test_non_stream_chat, "Non-streaming Chat Test")
|
773 |
run_test(test_stream_chat, "Streaming Chat Test")
|
774 |
run_test(test_query_modes, "Chat Query Mode Test")
|
775 |
+
|
|
|
|
|
776 |
if OutputControl.is_verbose():
|
777 |
print("\n【Generate API Tests】")
|
778 |
run_test(test_non_stream_generate, "Non-streaming Generate Test")
|
779 |
run_test(test_stream_generate, "Streaming Generate Test")
|
780 |
run_test(test_generate_with_system, "Generate with System Prompt Test")
|
|
|
781 |
run_test(test_generate_concurrent, "Generate Concurrent Test")
|
782 |
else:
|
783 |
# Run specified tests
|