diff --git a/src/bucket.py b/src/bucket.py index c4868b2..c0b25e7 100644 --- a/src/bucket.py +++ b/src/bucket.py @@ -54,3 +54,7 @@ class BucketChooserProportionalLogRandom: 1 + int(100 * math.log2(i[1])), ) for i in content ] + +class BucketChooserProportionalRandom(BucketChooserProportionalLogRandom): + def xform(self, content): + return [(i[0], 1+int(100*i[1])) for i in content] diff --git a/src/test_bucket.py b/src/test_bucket.py index 93ecb9a..11ef34a 100644 --- a/src/test_bucket.py +++ b/src/test_bucket.py @@ -34,6 +34,7 @@ class TestBucketChoosers(unittest.TestCase): def test_choosers(self): for chooser in [ bucket.BucketChooserProportionalLogRandom(), + bucket.BucketChooserProportionalRandom(), ]: with self.assertRaises(AssertionError): self.assertRaises(chooser.choose([]))