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)