@Test
    public void testGetShoulder() {
        Mass input = new Mass(4, 4, INPUT);
        Mass shoulder = new Mass(2, 8, SHOULDER);

        assertNull(graph.getShoulder());
        graph.addMasses(input, shoulder);
        assertSame(shoulder, graph.getShoulder());
    }