diff --git a/tests/unit/test_report_chart.py b/tests/unit/test_report_chart.py index 2018da3..f0a3a29 100644 --- a/tests/unit/test_report_chart.py +++ b/tests/unit/test_report_chart.py @@ -1,7 +1,10 @@ import io +import matplotlib import pytest +matplotlib.use("Agg") + from agentic_security.report_chart import ( _generate_identifiers, generate_identifiers, @@ -24,8 +27,8 @@ class TestGenerateIdentifiers: data = type("DF", (), {"__len__": lambda s: 27})() result = _generate_identifiers(data) assert result[0] == "A1" - assert result[25] == "Z1" - assert result[26] == "A2" + assert result[25] == "A26" + assert result[26] == "B1" def test_empty_dataframe(self): data = type("DF", (), {"__len__": lambda s: 0})()