def test_hierarchies(): activity_stats = ActivityStats() activity_stats.element = etree.fromstring(''' <iati-activity> </iati-activity> ''') assert activity_stats.hierarchies() == {None: 1} activity_stats = ActivityStats() activity_stats.element = etree.fromstring(''' <iati-activity hierarchy="3"> </iati-activity> ''') assert activity_stats.hierarchies() == {'3': 1}