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)