Если этот класс использует стороннюю библиотеку для создания графиков / диаграмм (скажем, matplotlib ), то вы можете написать тесты для методов, которые генерируют входные данные для библиотеки. Это будет довольно легко.
Если на выходе получается изображение и вы заинтересованы в проверке его свойств, вам придется копать глубже. Внешние атрибуты изображения (размер, высота, формат и т. Д.) Могут быть легко проверены, но другие, такие как фактическое содержание изображения, будут довольно сложными. ИМХО, это не стоило бы хлопот.
Если выходные данные не являются двоичными (скажем, SVG), то вы можете легко написать тесты, чтобы убедиться, что выходной XML содержит то, что вы ищете.