NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-07-12 10:07:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

Harden tooling dependency import tests

Browse Source
This commit is contained in:
Oleksandr Pavlyk
2026-06-29 12:59:54 -05:00
parent ee66a4d6c1
commit fd840b427b
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
python/test/test_nvbench_tooling_deps.py
View File

@@ -29,10 +29,17 @@ def test_tooling_deps_imports_from_packaged_script_path(tmp_path, monkeypatch):
)
monkeypatch.syspath_prepend(str(tmp_path))
sys.modules.pop("cuda.bench.scripts.nvbench_tooling_deps", None)
for module_name in [
"cuda",
"cuda.bench",
"cuda.bench.scripts",
"cuda.bench.scripts.nvbench_tooling_deps",
]:
monkeypatch.delitem(sys.modules, module_name, raising=False)
module = importlib.import_module("cuda.bench.scripts.nvbench_tooling_deps")
assert Path(module.__file__) == package_dir / "nvbench_tooling_deps.py"
assert module.ToolingDependency("math", "math", "testing").extra == "tools"