def test_parse_map_input0(self): path = os.getcwd() + '/input1.txt' arr = parseInputFile(path) myMap = makeMap(arr) newMap, count = parseMap(myMap) self.assertEqual(len(newMap), 11) self.assertEqual(count, 42) self.assertEqual(newMap['H']['total'], 3)
def test_parse_map(self): myMap = {'B': 'COM', 'C': 'B'} newMap, count = parseMap(myMap) expect = { 'B': { 'direct': 'COM', 'total': 1 }, 'C': { 'direct': 'B', 'total': 2 } } self.assertEqual(newMap, expect) self.assertEqual(count, 3)