yangdx commited on
Commit
153eff6
·
1 Parent(s): a8fc721

Remove error handling tests from "all" mode

Browse files
Files changed (1) hide show
  1. test_lightrag_ollama_chat.py +3 -6
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
- run_test(test_error_handling, "Chat Error Handling Test")
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