def test_get_frequencies(self): counters.increment(TEST_KEY, 'fred') counters.increment(TEST_KEY, 'barney') counters.increment(TEST_KEY, 'george') counters.increment(TEST_KEY, 'roger') counters.increment(TEST_KEY, 'roger') counters.increment(TEST_KEY, 'roger') counters.increment(TEST_KEY, 'roger') self.assertEqual(counters.get_frequencies(TEST_KEY), {1: 3, 4: 1})
def test_clear_value(self): counters.increment(TEST_KEY, 'fred') counters.increment(TEST_KEY, 'fred') counters.increment(TEST_KEY, 'fred') counters.increment(TEST_KEY, 'barney') counters.increment(TEST_KEY, 'barney') counters.clear(TEST_KEY, 'fred') self.assertEqual(counters.get(TEST_KEY), 1) self.assertEqual(counters.get_frequencies(TEST_KEY), {2: 1})
def goal_distribution(self, alternative, goal): return counters.get_frequencies(GOAL_KEY % (self.name, alternative, goal))