Fix: assert site-packages or dist-packages only if have_nvidia_nvshmem_package()

Author: rwgk

cuda_pathfinder/tests/test_find_nvidia_headers.py

@@ -44,11 +44,12 @@ def test_find_libname_nvshmem(info_summary_append):
     info_summary_append(f"{hdr_dir=!r}")
     if STRICTNESS == "all_must_work" or have_nvidia_nvshmem_package():
         assert hdr_dir is not None
-        hdr_dir_parts = hdr_dir.split(os.path.sep)
-        assert any(
-            sub_dir in hdr_dir_parts
-            for sub_dir in (
-                "site-packages",  # pip install
-                "dist-packages",  # apt install
-            )
-        ), hdr_dir
+        if have_nvidia_nvshmem_package():
+            hdr_dir_parts = hdr_dir.split(os.path.sep)
+            assert any(
+                sub_dir in hdr_dir_parts
+                for sub_dir in (
+                    "site-packages",  # pip install
+                    "dist-packages",  # apt install
+                )
+            ), hdr_dir