Skip to content

Commit b74a84c

Browse files
committed
Additional testing
1 parent 2426260 commit b74a84c

1 file changed

Lines changed: 7 additions & 0 deletions

File tree

cuda_pathfinder/tests/test_find_nvidia_headers.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,8 +46,15 @@ def test_find_libname_nvshmem(info_summary_append):
4646
if IS_WINDOWS:
4747
assert hdr_dir is None
4848
pytest.skip("nvshmem has no Windows support.")
49+
if hdr_dir:
50+
assert os.path.isdir(hdr_dir)
51+
assert os.path.isfile(os.path.join(hdr_dir, "nvshmem.h"))
4952
if STRICTNESS == "all_must_work" or have_nvidia_nvshmem_package():
5053
assert hdr_dir is not None
5154
if have_nvidia_nvshmem_package():
5255
hdr_dir_parts = hdr_dir.split(os.path.sep)
5356
assert "site-packages" in hdr_dir_parts
57+
elif conda_prefix := os.getenv("CONDA_PREFIX"):
58+
assert hdr_dir.startswith(conda_prefix)
59+
else:
60+
assert hdr_dir.startswith("/usr/include/nvshmem_")

0 commit comments

Comments
 (0)