@Test public void testGraph() throws Exception { Graph graph = new PeopleGraphImpl().graphFor(1); graph.add(new PersonImpl(1, "Zequinha", 10.0)); graph.add(new PersonImpl(2, "Barbosa", 20.0)); List<Person> closest = graph.getClosestFriends(5); assertEquals(2, closest.size()); }
@Test public void graphWithTwoNodesAndTwoNeighbours() throws Exception { Node node = new Node("1"); node.addNeighbour(new Node("3")); Node node2 = new Node("2"); node2.addNeighbour(new Node("4")); graph.add(node); graph.add(node2); assertEquals("1 3\n2 4\n", serializeGraph()); }
@Test public void graphWithOneNodeAndOneNeighbourReturnsNodeAndNeighbour() throws Exception { Node node = new Node("1"); node.addNeighbour(new Node("2")); graph.add(node); assertEquals("1 2\n", serializeGraph()); }
private void addNode(String nodeName) { graph.add(new Node(nodeName)); }