Update Diagram tests for Luna