summaryrefslogtreecommitdiff
path: root/tests/test_cuda_setup_evaluator.py
diff options
context:
space:
mode:
authorTitus von Koeller <titus@vonkoeller.com>2022-08-01 09:32:47 -0700
committerTitus von Koeller <titus@vonkoeller.com>2022-08-01 09:32:47 -0700
commitea7c14f8ef64924f2d0ff80df3cdabf2c7299848 (patch)
tree3b9ec443a259cf36d87627a8e2cc7d13513f6a21 /tests/test_cuda_setup_evaluator.py
parent3fd06fb6206f46b6d18fbb8a512da63832dea98b (diff)
reran black with linelength 80 for greater readability
Diffstat (limited to 'tests/test_cuda_setup_evaluator.py')
-rw-r--r--tests/test_cuda_setup_evaluator.py33
1 files changed, 26 insertions, 7 deletions
diff --git a/tests/test_cuda_setup_evaluator.py b/tests/test_cuda_setup_evaluator.py
index d45354f..5a58be4 100644
--- a/tests/test_cuda_setup_evaluator.py
+++ b/tests/test_cuda_setup_evaluator.py
@@ -3,8 +3,12 @@ from typing import List, NamedTuple
import pytest
-from bitsandbytes.cuda_setup import (CUDA_RUNTIME_LIB, evaluate_cuda_setup,
- get_cuda_runtime_lib_path, tokenize_paths)
+from bitsandbytes.cuda_setup import (
+ CUDA_RUNTIME_LIB,
+ evaluate_cuda_setup,
+ get_cuda_runtime_lib_path,
+ tokenize_paths,
+)
class InputAndExpectedOutput(NamedTuple):
@@ -13,11 +17,26 @@ class InputAndExpectedOutput(NamedTuple):
HAPPY_PATH__LD_LIB_TEST_PATHS: List[InputAndExpectedOutput] = [
- (f"some/other/dir:dir/with/{CUDA_RUNTIME_LIB}", f"dir/with/{CUDA_RUNTIME_LIB}"),
- (f":some/other/dir:dir/with/{CUDA_RUNTIME_LIB}", f"dir/with/{CUDA_RUNTIME_LIB}"),
- (f"some/other/dir:dir/with/{CUDA_RUNTIME_LIB}:", f"dir/with/{CUDA_RUNTIME_LIB}"),
- (f"some/other/dir::dir/with/{CUDA_RUNTIME_LIB}", f"dir/with/{CUDA_RUNTIME_LIB}"),
- (f"dir/with/{CUDA_RUNTIME_LIB}:some/other/dir", f"dir/with/{CUDA_RUNTIME_LIB}"),
+ (
+ f"some/other/dir:dir/with/{CUDA_RUNTIME_LIB}",
+ f"dir/with/{CUDA_RUNTIME_LIB}",
+ ),
+ (
+ f":some/other/dir:dir/with/{CUDA_RUNTIME_LIB}",
+ f"dir/with/{CUDA_RUNTIME_LIB}",
+ ),
+ (
+ f"some/other/dir:dir/with/{CUDA_RUNTIME_LIB}:",
+ f"dir/with/{CUDA_RUNTIME_LIB}",
+ ),
+ (
+ f"some/other/dir::dir/with/{CUDA_RUNTIME_LIB}",
+ f"dir/with/{CUDA_RUNTIME_LIB}",
+ ),
+ (
+ f"dir/with/{CUDA_RUNTIME_LIB}:some/other/dir",
+ f"dir/with/{CUDA_RUNTIME_LIB}",
+ ),
(
f"dir/with/{CUDA_RUNTIME_LIB}:other/dir/libcuda.so",
f"dir/with/{CUDA_RUNTIME_LIB}",