44
55import pytest
66
7+ from cuda .core ._utils import enum_explanations_helpers
78from cuda .core ._utils .enum_explanations_helpers import (
89 DocstringBackedExplanations ,
910 _binding_version_has_usable_enum_docstrings ,
@@ -74,16 +75,6 @@ def test_docstring_backed_get_returns_default_for_missing_docstring():
7475 assert lut .get (7 , default = "sentinel" ) == "sentinel"
7576
7677
77- def test_docstring_backed_get_returns_default_for_unknown_code ():
78- lut = DocstringBackedExplanations (_FakeEnumType ({}))
79- assert lut .get (99 , default = "sentinel" ) == "sentinel"
80-
81-
82- def test_docstring_backed_get_returns_default_for_missing_docstring_without_fallback ():
83- lut = DocstringBackedExplanations (_FakeEnumType ({7 : _FakeEnumMember (None )}))
84- assert lut .get (7 , default = "sentinel" ) == "sentinel"
85-
86-
8778@pytest .mark .parametrize (
8879 ("version" , "expected" ),
8980 [
@@ -108,11 +99,9 @@ def test_binding_version_has_usable_enum_docstrings(version, expected):
10899 ],
109100)
110101def test_get_best_available_explanations_switches_by_version (monkeypatch , version , expects_docstrings ):
111- import cuda .core ._utils .enum_explanations_helpers as cleanup
112-
113102 fallback = {7 : "fallback text" }
114- monkeypatch .setattr (cleanup , "_binding_version" , lambda : version )
115- expl = cleanup .get_best_available_explanations (
103+ monkeypatch .setattr (enum_explanations_helpers , "_binding_version" , lambda : version )
104+ expl = enum_explanations_helpers .get_best_available_explanations (
116105 _FakeEnumType ({7 : _FakeEnumMember ("clean me" )}),
117106 fallback ,
118107 )
@@ -121,13 +110,3 @@ def test_get_best_available_explanations_switches_by_version(monkeypatch, versio
121110 assert expl .get (7 ) == "clean me"
122111 else :
123112 assert expl is fallback
124-
125-
126- def test_driver_cu_result_explanations_get_matches_clean_docstring ():
127- pytest .importorskip ("cuda.bindings" )
128- from cuda .bindings import driver
129- from cuda .core ._utils .driver_cu_result_explanations import DRIVER_CU_RESULT_EXPLANATIONS
130-
131- e = driver .CUresult .CUDA_SUCCESS
132- code = int (e )
133- assert DRIVER_CU_RESULT_EXPLANATIONS .get (code ) == clean_enum_member_docstring (e .__doc__ )
0 commit comments