def test_parent_child_stop_point(self):
xml_handler = NaptanXMLHandler(['639'], 'identifiers')
parseString(test_stop_areas, xml_handler)
areas = xml_handler.annotate_stop_area_ancestry(xml_handler.stop_areas)
points, areas = xml_handler.annotate_stop_point_ancestry(xml_handler.stop_points, areas)
self.assertEqual(points['639000022']['child_of'][0], areas['639GSHI21581']['id'])
self.assertEqual(areas['639GSHI21581']['parent_of'][0], points['639000022']['id'])
def test_parent_child_stop_areas_inactive(self):
xml_handler = NaptanXMLHandler(['639'], 'identifiers')
parseString(test_stop_areas, xml_handler)
areas = xml_handler.annotate_stop_area_ancestry(xml_handler.stop_areas)
with self.assertRaises(KeyError):
self.assertEqual(areas['639GSHI21580']['child_of'][0], areas['639GSHI20121']['id'])