def test_by_hierarchy(): # Unlike the hierarchies dict, by_hierarchy should treat activities without # a hierarchy attribute as hierarchy 1 activity_stats = ActivityStats() activity_stats.element = etree.fromstring(''' <iati-activity> </iati-activity> ''') assert activity_stats.by_hierarchy().keys() == [ '1' ] activity_stats = ActivityStats() activity_stats.element = etree.fromstring(''' <iati-activity hierarchy="3"> </iati-activity> ''') assert activity_stats.by_hierarchy().keys() == [ '3' ]
def test_by_hierarchy(): # Unlike the hierarchies dict, by_hierarchy should treat activities without # a hierarchy attribute as hierarchy 1 activity_stats = ActivityStats() activity_stats.element = etree.fromstring(''' <iati-activity> </iati-activity> ''') assert activity_stats.by_hierarchy().keys() == ['1'] activity_stats = ActivityStats() activity_stats.element = etree.fromstring(''' <iati-activity hierarchy="3"> </iati-activity> ''') assert activity_stats.by_hierarchy().keys() == ['3']