Make tests easier to debug

rapidsai · Mar 13, 2024 · 7dce611 · 7dce611
1 parent d975362
commit 7dce611
Showing 1 changed file with 26 additions and 21 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -121,27 +121,32 @@ def install(self, *args):
         )
 
     def wheel(self, package_dir, *args):
-        return subprocess.run(
-            [
-                self.executable,
-                "-m",
-                "pip",
-                "wheel",
-                "--disable-pip-version-check",
-                "--no-deps",
-                "--wheel-dir",
-                package_dir,
-                "--find-links",
-                str(self.wheelhouse),
-                "--cache-dir",
-                self.cache_dir,
-                package_dir,
-                *args,
-            ],
-            stdout=subprocess.PIPE,
-            stderr=subprocess.STDOUT,
-            check=True,
-        )
+        try:
+            return subprocess.run(
+                [
+                    self.executable,
+                    "-m",
+                    "pip",
+                    "wheel",
+                    "--disable-pip-version-check",
+                    "--no-deps",
+                    "--wheel-dir",
+                    package_dir,
+                    "--find-links",
+                    str(self.wheelhouse),
+                    "--cache-dir",
+                    self.cache_dir,
+                    package_dir,
+                    *args,
+                ],
+                stdout=subprocess.PIPE,
+                stderr=subprocess.STDOUT,
+                check=True,
+            )
+        except subprocess.CalledProcessError as e:
+            print("Subprocess failed with output:")
+            print(e.stdout.decode("utf-8"))
+            raise
 
 
 @pytest.fixture(scope="session")