test: remove B095 ruff rule and add zip(..., strict=True) in test code

Author: cpcloud

cuda_bindings/setup.py

@@ -125,7 +125,7 @@ def discoverMembers(self, memberDict, prefix, seen=None):
         next_seen = set(seen)
         next_seen.add(self._name)
 
-        for memberName, memberType in zip(self._member_names, self._member_types):
+        for memberName, memberType in zip(self._member_names, self._member_types, strict=True):
             if memberName:
                 discovered.append(".".join([prefix, memberName]))
 

cuda_bindings/tests/test_cuda.py

@@ -432,7 +432,7 @@ def test_cuda_pointer_attr():
     # List version
     err, attr_value_list_v2 = cuda.cuPointerGetAttributes(len(attr_type_list), attr_type_list, ptr)
     assert err == cuda.CUresult.CUDA_SUCCESS
-    for attr1, attr2 in zip(attr_value_list, attr_value_list_v2):
+    for attr1, attr2 in zip(attr_value_list, attr_value_list_v2, strict=True):
         assert str(attr1) == str(attr2)
 
     # Test setting values
@@ -512,7 +512,7 @@ def test_cuda_mem_range_attr():
         attr_type_size_list, attr_type_list, len(attr_type_list), ptr, size
     )
     assert err == cuda.CUresult.CUDA_SUCCESS
-    for attr1, attr2 in zip(attr_value_list, attr_value_list_v2):
+    for attr1, attr2 in zip(attr_value_list, attr_value_list_v2, strict=True):
         assert str(attr1) == str(attr2)
 
     (err,) = cuda.cuMemFree(ptr)

cuda_bindings/tests/test_cufile.py

@@ -318,7 +318,7 @@ def test_buf_register_multiple_buffers():
     try:
         # Register all buffers
         flags = 0
-        for buf_ptr, size in zip(buffers, buffer_sizes):
+        for buf_ptr, size in zip(buffers, buffer_sizes, strict=True):
             buf_ptr_int = int(buf_ptr)
             cufile.buf_register(buf_ptr_int, size, flags)
 

cuda_bindings/tests/test_nvjitlink.py

@@ -34,7 +34,7 @@
 
 
 def _build_arch_ptx_parametrized_callable():
-    av = tuple(zip(ARCHITECTURES, PTX_VERSIONS))
+    av = tuple(zip(ARCHITECTURES, PTX_VERSIONS, strict=True))
     return pytest.mark.parametrize(
         ("arch", "ptx_bytes"),
         [(a, (PTX_HEADER.format(VERSION=v, ARCH=a) + PTX_KERNEL).encode("utf-8")) for a, v in av],

cuda_core/tests/test_module.py

@@ -248,9 +248,9 @@ class ExpectedStruct(ctypes.Structure):
     sizes = [p.size for p in arg_info]
     members = [getattr(ExpectedStruct, name) for name, _ in ExpectedStruct._fields_]
     expected_offsets = tuple(m.offset for m in members)
-    assert all(actual == expected for actual, expected in zip(offsets, expected_offsets))
+    assert all(actual == expected for actual, expected in zip(offsets, expected_offsets, strict=True))
     expected_sizes = tuple(m.size for m in members)
-    assert all(actual == expected for actual, expected in zip(sizes, expected_sizes))
+    assert all(actual == expected for actual, expected in zip(sizes, expected_sizes, strict=True))
 
 
 @pytest.mark.parametrize("nargs", [0, 1, 2, 3, 16])
@@ -274,8 +274,8 @@ class ExpectedStruct(ctypes.Structure):
     members = tuple(getattr(ExpectedStruct, f"arg_{i}") for i in range(nargs))
 
     arg_info = krn.arguments_info
-    assert all([actual.offset == expected.offset for actual, expected in zip(arg_info, members)])
-    assert all([actual.size == expected.size for actual, expected in zip(arg_info, members)])
+    assert all([actual.offset == expected.offset for actual, expected in zip(arg_info, members, strict=True)])
+    assert all([actual.size == expected.size for actual, expected in zip(arg_info, members, strict=True)])
 
 
 def test_num_args_error_handling(deinit_all_contexts_function, cuda12_4_prerequisite_check):

cuda_core/tests/test_system.py

@@ -35,5 +35,5 @@ def test_devices():
     expected_num_devices = handle_return(runtime.cudaGetDeviceCount())
     expected_devices = tuple(Device(device_id) for device_id in range(expected_num_devices))
     assert len(devices) == len(expected_devices), "Number of devices does not match expected value"
-    for device, expected_device in zip(devices, expected_devices):
+    for device, expected_device in zip(devices, expected_devices, strict=True):
         assert device.device_id == expected_device.device_id, "Device ID does not match expected value"

ruff.toml

@@ -40,7 +40,6 @@ ignore = [
   "S101",   # asserts
   "S311",   # allow use of the random.* even though many are not cryptographically secure
   "S404",   # allow importing the subprocess module
-  "B905",   # preserve the default behavior of `zip` without the explicit `strict` argument
 ]
 
 exclude = ["**/_version.py"]
@@ -52,6 +51,7 @@ exclude = ["**/_version.py"]
 "cuda_bindings/examples/**" = [
   "E722",
   "E501", # line too long
+  "B905",   # preserve the default behavior of `zip` without the explicit `strict` argument
 ]
 
 "cuda_bindings/tests/**" = [