TestDijkstraShortestPath.java 文件源码

private void assertShortestPath(String[] input) {
    for (String line : input) {
        String[] values = line.split(" ");
        switch (values[0]) {
            case "compute":
                dijkstra.computeShortestPath(values[1]);
                break;
            case "shortestPath":
                String[] expected = values[2].split(",");
                if (expected[0].equals("empty")) {
                    assertThat(dijkstra.getShortestPathTo(values[1]), IsEmptyCollection.empty());
                } else {
                    assertThat(dijkstra.getShortestPathTo(values[1]), contains(expected));
                }
                break;
            case "shortestDistance":
                Integer expectedDistance = Integer.MAX_VALUE;
                if (!values[2].equals("infinity")) {
                    expectedDistance = Integer.parseInt(values[2]);
                }
                assertThat(dijkstra.getShortestDistanceTo(values[1]), is(expectedDistance));
                break;
        }
    }
}