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= compute.sumFunctionalWaterPoints(dataSet)
, number_water_points = compute.sumWaterPointsPerSplitKey("communities_villages", dataSet)
, community_ranking = compute.rank("water_functioning", "communities_villages", dataSet)
)
def test_sumWaterPointsPerCommunity(self):
assert compute.sumWaterPointsPerSplitKey("communities_villages", TestCompute.data) == dict(VilA=2, VilB=2, VilC=1)