@Test
    public void testGetInputNodes() {
        mass1 = new Mass(1, 2, INPUT);
        mass2 = new Mass(3, 4);
        graph.addMasses(mass1, mass2);

        assertEquals(1, graph.getInputs().size());
        assertSame(mass1, graph.getInputs().get(0));
    }