further unit tests for diagram