Add tests for get diagram methods.
Add dummy diagrams for test cases.