@Test public void smallGraph() { Node A = new UndirectedGraphNode("A"); Node B = new UndirectedGraphNode("B"); Node C = new UndirectedGraphNode("C"); Node D = new UndirectedGraphNode("D"); Node E = new UndirectedGraphNode("E"); test.addNode(A); test.addNode(B); test.addNode(C); test.addNode(D); test.addNode(E); test.addEdge(A, B); assertEquals(Arrays.asList(B), A.getNeighbours()); test.addEdge(A, C); assertEquals(Arrays.asList(B, C), A.getNeighbours()); test.addEdge(B, D); assertEquals(Arrays.asList(A, D), B.getNeighbours()); test.addEdge(C, D); assertEquals(Arrays.asList(A, D), C.getNeighbours()); test.addEdge(D, E); assertEquals(Arrays.asList(B, C, E), D.getNeighbours()); assertEquals(Arrays.asList(D), E.getNeighbours()); }