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

Test to use --bulk-debug-python stdout, not STDOUT

Browse Source 
Also add a test to check that STDOUT also works.
This commit is contained in:
Oleksandr Pavlyk
2026-06-28 07:26:08 -05:00
parent 2e0f635b6c
commit adb7e2a0ca
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
python/test/test_nvbench_compare.py
View File

@@ -655,6 +655,12 @@ def test_format_bulk_debug_python_loads_arrays(tmp_path, nvbench_compare):
assert arrays["reference_frequencies"] is None
def test_bulk_debug_output_stdout_token_is_case_insensitive(nvbench_compare):
assert nvbench_compare.BulkDebugOutput("stdout").is_stdout
assert nvbench_compare.BulkDebugOutput("STDOUT").is_stdout
assert not nvbench_compare.BulkDebugOutput("output.py").is_stdout
def test_format_bulk_debug_python_handles_nonfinite_values(nvbench_compare):
script = nvbench_compare.format_bulk_debug_python(
[
@@ -2611,7 +2617,7 @@ def test_main_prints_bulk_debug_python_to_stdout(monkeypatch, capsys, nvbench_co
[
"nvbench_compare",
"--bulk-debug-python",
"STDOUT",
"stdout",
"ref.json",
"cmp.json",
],