def generateDict (dataSet: dict) -> dict:
"""
For testing on a smaller dataset and increased modularity.
Returns a data structure with the following format:
{
number_functional: …,
number_water_points: {
communityA: …,
},
community_ranking: …
}
"""
return dict( number_functional= ci.sumFunctionalWaterPoints(dataSet)
, number_water_points = ci.sumWaterPointsPerCommunity(dataSet)
, community_ranking = ci.rankCommunitiesByPercentBrokenWaterPoints(dataSet)
)
def test_sumWaterPointsPerCommunity(self):
self.assertEqual(ci.sumWaterPointsPerCommunity(TestComputeWaterInfrastructure.dataSet)
, dict(VilA=2, VilB=2, VilC=1))