simplifying

This commit is contained in:
Dan Saunders
2025-02-28 10:19:40 -05:00
committed by Dan Saunders
parent 2d36c11264
commit 035e7a2f4c
2 changed files with 16 additions and 16 deletions

View File

@@ -106,11 +106,12 @@ def test_telemetry_disabled_for_non_main_process(telemetry_manager_class):
def test_opt_in_info_displayed(telemetry_manager_class):
"""Test that opt-in info is displayed when telemetry is not configured"""
with patch.dict(os.environ, {"RANK": "0"}, clear=True), patch(
"logging.Logger.info"
) as mock_info, patch("time.sleep"):
"logging.Logger.warning"
) as mock_warning, patch("time.sleep"):
telemetry_manager_class()
info_displayed = False
for call in mock_info.call_args_list:
for call in mock_warning.call_args_list:
print(f"call: {call}")
if "Telemetry is currently disabled by default" in str(call):
info_displayed = True
break