def test_get_random_weighted_item(self): dsg = Dictogram(self.words) dsg.update(('c',)) random_item = dsg._get_random_item_by_weight(2) self.assertEqual(random_item[0], 'c') dsg.update(('b',)) dsg.update(('c',)) random_item = dsg._get_random_item_by_weight(3) self.assertEqual(random_item[0], 'c')
def test_update_2(self): dsg = Dictogram(self.words) dsg.update(self.words) self.assertEquals({'a': 2, 'b': 2, 'c': 2}, dsg) dsg.update(('c', 'b',)) self.assertEquals({'a': 2, 'b': 3, 'c': 3}, dsg) dsg.update(('b',)) self.assertEquals({'a': 2, 'b': 4, 'c': 3}, dsg)
def test_update_1(self): dsg = Dictogram() dsg.update(self.words) self.assertEquals({'a': 1, 'b': 1, 'c': 1}, dsg)