Add test for Image.generate_diagram