Add test for Container.generate_diagram