Skip to content

Commit e8b34ec

Browse files
committed
Fix test on Py < 3.12
1 parent 6cbea13 commit e8b34ec

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

cuda_bindings/tests/test_basics.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
# SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
33

44
import enum
5+
import sys
56

67
import pytest
78
from cuda.bindings.utils import _fast_enum
@@ -59,7 +60,8 @@ def test_enum(MyEnum):
5960

6061
for item in container:
6162
assert isinstance(item, container)
62-
assert item in container
63+
if sys.version_info >= (3, 12):
64+
assert item in container
6365
assert item.value in container
6466
assert item.name in dir(container)
6567
for item2 in container:

0 commit comments

Comments
 (0)