chore: better test logs

Author: lmxyy

.github/workflows/pr-test.yaml

@@ -108,13 +108,13 @@ jobs:
           pwd
           cd ../ComfyUI
           python nunchaku_tests/scripts/nunchaku-flux1-dev.py
-          pytest -s nunchaku_tests/
+          pytest -s -x nunchaku_tests/
       - name: Run nunchaku tests
         run: |
           source $(conda info --base)/etc/profile.d/conda.sh
           conda activate test_env || { echo "Failed to activate conda env"; exit 1; }
           which python
-          pytest -s tests/flux/test_flux_examples.py
+          pytest -s -x tests/flux/test_flux_examples.py
           python .github/workflows/run_all_tests.py
       - name: clean up
         if: always()

.github/workflows/run_all_tests.py

@@ -21,7 +21,7 @@ def run_all_tests():
     failed_tests = []
     for test_file in test_files:
         print(f"Running {test_file} ...")
-        result = subprocess.run(["pytest", "--reruns", "2", "--reruns-delay", "0", "-s", test_file])
+        result = subprocess.run(["pytest", "--reruns", "2", "--reruns-delay", "0", "-s", "-x", test_file])
         if result.returncode != 0:
             print(f"Test failed: {test_file}")
             failed_tests.append(test_file)

tests/flux/test_flux_examples.py

@@ -12,9 +12,7 @@
 
 @pytest.mark.parametrize("script_name", example_scripts)
 def test_example_script_runs(script_name):
-    gc.collect()
-    torch.cuda.empty_cache()
     script_path = os.path.join(EXAMPLES_DIR, script_name)
-    result = subprocess.run(["python", script_path], capture_output=True, text=True)
+    result = subprocess.run(["python", script_path], text=True)
     print(f"Running {script_path} -> Return code: {result.returncode}")
     assert result.returncode == 0, f"{script_path} failed with code {result.returncode}"