Python parse_in_file示例

示例#1

文件： test.py 项目： SHLo/algorithms

    def test_sample_1(self):
        answer = {
            '1': {
                'length': 0,
                'path': []
            },
            '2': {
                'length': 3,
                'path': ['2']
            },
            '3': {
                'length': 3,
                'path': ['3']
            },
            '4': {
                'length': 5,
                'path': ['2', '4']
            }
        }

        graph = dijkstra.parse_in_file('sample-1.txt')
        output = dijkstra.dijkstra(graph)
        self.assertEqual(answer, output)

示例#2

文件： test.py 项目： SHLo/algorithms

    def test_medium_size(self):
        answer = 5

        graph = dijkstra.parse_in_file('medium-size.txt')
        output = dijkstra.dijkstra(graph)
        self.assertEqual(answer, output['7']['length'])