style: linting

Author: cpcloud

cuda_pathfinder/tests/test_ctk_root_discovery.py

@@ -6,9 +6,9 @@
 import pytest
 
 from cuda.pathfinder._dynamic_libs.find_nvidia_dynamic_lib import (
-    _FindNvidiaDynamicLib,
     _derive_ctk_root_linux,
     _derive_ctk_root_windows,
+    _FindNvidiaDynamicLib,
     derive_ctk_root,
 )
 from cuda.pathfinder._dynamic_libs.load_dl_common import LoadedDL
@@ -17,7 +17,6 @@
     _try_ctk_root_canary,
 )
 
-
 # ---------------------------------------------------------------------------
 # derive_ctk_root
 # ---------------------------------------------------------------------------
@@ -203,20 +202,19 @@ def test_cuda_home_takes_priority_over_canary(tmp_path):
     nvvm_canary.mkdir(parents=True)
     (nvvm_canary / "libnvvm.so").write_bytes(b"canary")
 
-    canary_mock = MagicMock(return_value=_make_loaded_dl(
-        str(canary_root / "lib64" / "libcudart.so.13"), "system-search"
-    ))
+    canary_mock = MagicMock(
+        return_value=_make_loaded_dl(str(canary_root / "lib64" / "libcudart.so.13"), "system-search")
+    )
 
     with (
         # System search finds nothing for nvvm; canary would find cudart
-        patch(f"{_MODULE}.load_with_system_search",
-              side_effect=lambda name: None if name == "nvvm" else canary_mock(name)),
+        patch(
+            f"{_MODULE}.load_with_system_search", side_effect=lambda name: None if name == "nvvm" else canary_mock(name)
+        ),
         # CUDA_HOME points to a separate root that also has nvvm
-        patch(f"{_FIND_MODULE}.get_cuda_home_or_path",
-              return_value=str(cuda_home_root)),
+        patch(f"{_FIND_MODULE}.get_cuda_home_or_path", return_value=str(cuda_home_root)),
         # Capture the final load call
-        patch(f"{_MODULE}.load_with_abs_path",
-              side_effect=lambda libname, path, via: _make_loaded_dl(path, via)),
+        patch(f"{_MODULE}.load_with_abs_path", side_effect=lambda libname, path, via: _make_loaded_dl(path, via)),
     ):
         result = _load_lib_no_cache("nvvm")
 
@@ -235,19 +233,17 @@ def test_canary_fires_only_after_all_earlier_steps_fail(tmp_path):
     nvvm_so = nvvm_dir / "libnvvm.so"
     nvvm_so.write_bytes(b"canary")
 
-    canary_result = _make_loaded_dl(
-        str(canary_root / "lib64" / "libcudart.so.13"), "system-search"
-    )
+    canary_result = _make_loaded_dl(str(canary_root / "lib64" / "libcudart.so.13"), "system-search")
 
     with (
         # System search: nvvm not on linker path, but cudart (canary) is
-        patch(f"{_MODULE}.load_with_system_search",
-              side_effect=lambda name: canary_result if name == "cudart" else None),
+        patch(
+            f"{_MODULE}.load_with_system_search", side_effect=lambda name: canary_result if name == "cudart" else None
+        ),
         # No CUDA_HOME set
         patch(f"{_FIND_MODULE}.get_cuda_home_or_path", return_value=None),
         # Capture the final load call
-        patch(f"{_MODULE}.load_with_abs_path",
-              side_effect=lambda libname, path, via: _make_loaded_dl(path, via)),
+        patch(f"{_MODULE}.load_with_abs_path", side_effect=lambda libname, path, via: _make_loaded_dl(path, via)),
     ):
         result = _load_lib_no_cache("nvvm")