From 550154c9c445498c468ba60d39e5c4cfdd8ea0fb Mon Sep 17 00:00:00 2001
From: Oleksandr Pavlyk <21087696+oleksandr-pavlyk@users.noreply.github.com>
Date: Tue, 14 Jul 2026 15:38:05 -0500
Subject: [PATCH] Tighten nvbench_compare test coverage

---
 python/test/test_nvbench_compare.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/python/test/test_nvbench_compare.py b/python/test/test_nvbench_compare.py
index 4cd87fd..f8a582e 100644
--- a/python/test/test_nvbench_compare.py
+++ b/python/test/test_nvbench_compare.py
@@ -2952,6 +2952,7 @@ def test_main_validates_device_filters_before_loading_tooling(
 
 def test_main_no_color_does_not_require_colorama(monkeypatch, nvbench_compare):
     calls = []
+    monkeypatch.setattr(nvbench_compare, "Fore", None)
 
     def fake_load_tooling(*, load_color=True):
         calls.append(load_color)
@@ -2981,6 +2982,7 @@ def test_main_no_color_does_not_require_colorama(monkeypatch, nvbench_compare):
 
     assert nvbench_compare.main() == 0
     assert calls == [False]
+    assert nvbench_compare.Fore is None
 
 
 def test_main_reports_input_read_failures(monkeypatch, capsys, nvbench_compare):