Skip to content

Commit 614dead

Browse files
committed
Test both bitcode lookup keyword-only APIs
1 parent e3ab8f5 commit 614dead

1 file changed

Lines changed: 9 additions & 2 deletions

File tree

cuda_pathfinder/tests/test_find_bitcode_lib.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -293,9 +293,16 @@ def test_find_bitcode_lib_invalid_name():
293293
find_bitcode_lib_module.locate_bitcode_lib("invalid")
294294

295295

296-
def test_find_bitcode_lib_sm_arch_is_keyword_only():
296+
@pytest.mark.parametrize(
297+
"find_fn",
298+
[
299+
find_bitcode_lib,
300+
locate_bitcode_lib,
301+
],
302+
)
303+
def test_bitcode_lib_sm_arch_is_keyword_only(find_fn):
297304
with pytest.raises(TypeError):
298-
find_bitcode_lib("nvshmem_device", "sm90")
305+
find_fn("nvshmem_device", "sm90")
299306

300307

301308
@pytest.mark.parametrize(

0 commit comments

Comments
 (0)