More fixes

Author: mdboom

cuda_core/tests/example_tests/test_basic_examples.py

@@ -5,6 +5,7 @@
 
 import glob
 import os
+import platform
 import re
 import subprocess
 import sys
@@ -18,8 +19,8 @@
 # 1) Directly running the example in the same environment as the test suite.
 #    This gives access to the current development version of cuda_core.
 # 2) Running the example in a subprocess with "uv run" to verify that the PEP
-#    723 metadata installation works correctly and that the example can be run in
-#    isolation from the test suite.
+#    723 metadata works correctly and that the example can be run in isolation from
+#    the test suite.
 
 
 def has_compute_capability_9_or_higher() -> bool:
@@ -36,6 +37,14 @@ def has_display() -> bool:
     return False
 
 
+def is_not_windows() -> bool:
+    return sys.platform != "win32"
+
+
+def is_x86_64() -> bool:
+    return platform.machine() == "x86_64"
+
+
 def uv_installed() -> bool:
     try:
         subprocess.run(["uv", "--version"], check=True)  # noqa: S607
@@ -45,10 +54,10 @@ def uv_installed() -> bool:
 
 
 PACKAGE_REQUIREMENTS = {
-    "pytorch_example.py": ["torch"],
     "cuda_graphs.py": ["cupy"],
     "jit_lto_fractal.py": ["cupy"],
     "memory_ops.py": ["cupy"],
+    "pytorch_example.py": ["torch"],
     "saxpy.py": ["cupy"],
     "simple_multi_gpu_example.py": ["cupy"],
     "strided_memory_view_cpu.py": ["cffi"],
@@ -60,8 +69,10 @@ def uv_installed() -> bool:
 
 SYSTEM_REQUIREMENTS = {
     "gl_interop_plasma.py": has_display,
-    "thread_block_cluster.py": has_compute_capability_9_or_higher,
+    "pytorch_example.py": is_x86_64,  # PyTorch only provides CUDA support for x86_64
     "simple_multi_gpu_example.py": has_multiple_devices,
+    "strided_memory_view_cpu.py": is_not_windows,
+    "thread_block_cluster.py": has_compute_capability_9_or_higher,
 }
 
 
@@ -86,11 +97,14 @@ def test_example(self, example):
         example_path = os.path.join(samples_path, example)
         process = subprocess.run([sys.executable, example_path], capture_output=True)  # noqa: S603
         if process.returncode != 0:
-            print(process.stdout.decode())
-            print(process.stderr.decode(), file=sys.stderr)
-            raise AssertionError(f"Example failed with return code {process.returncode} ({example})")
+            if process.stdout:
+                print(process.stdout.decode())
+            if process.stderr:
+                print(process.stderr.decode(), file=sys.stderr)
+            raise AssertionError(f"`{example}` failed ({process.returncode})")
 
     @pytest.mark.skipif(not uv_installed(), reason="uv is required to test PEP 723 metadata installation")
+    @pytest.mark.skipif(sys.platform == "win32", reason="uv does not currently work on Windows in CI")
     def test_example_pep723(self, example):
         system_requirement = SYSTEM_REQUIREMENTS.get(example, lambda: True)
         if not system_requirement():
@@ -101,9 +115,11 @@ def test_example_pep723(self, example):
         if process.returncode != 0:
             # This example requires a development version of cuda_core, so requirements can't be met.
             # That's ok, it was tested in the other test, so we just skip it instead of failing.
-            if re.match(process.stderr.decode(), "Because only cuda-(core|bindings)"):
+            if re.search("Because only cuda-(core)|(bindings)", process.stderr.decode()):
                 pytest.skip(f"Skipping {example} due to unmet PEP 723 requirement")
 
-            print(process.stdout.decode())
-            print(process.stderr.decode(), file=sys.stderr)
-            raise AssertionError(f"Example failed with return code {process.returncode} ({example})")
+            if process.stdout:
+                print(process.stdout.decode())
+            if process.stderr:
+                print(process.stderr.decode(), file=sys.stderr)
+            raise AssertionError(f"`uv run {example}` failed ({process.returncode})")