def test_distribution_single_bucket(self):
cfg = [
{
"name": "variant_1",
"size": 0.001
},
{
"name": "variant_2",
"size": 0
},
]
variant_set = SingleVariantSet(variants=cfg,
num_buckets=NUM_BUCKETS_DEFAULT)
variant_counts = {
"variant_1": 0,
"variant_2": 0,
None: 0,
}
for bucket in range(0, NUM_BUCKETS_DEFAULT):
variant = variant_set.choose_variant(bucket)
variant_counts[variant] += 1
self.assertEqual(len(variant_counts), 3)
self.assertEqual(variant_counts['variant_1'], 1)
self.assertEqual(variant_counts['variant_2'], 0)
self.assertEqual(variant_counts[None], 999)
def test_distribution_def_odd(self):
variant_cfg = [
{
"name": "variant_1",
"size": 0.5
},
{
"name": "variant_2",
"size": 0.5
},
]
variant_set = SingleVariantSet(
variants=variant_cfg,
num_buckets=NUM_BUCKETS_ODD,
)
variant_counts = {
"variant_1": 0,
"variant_2": 0,
None: 0,
}
for bucket in range(0, NUM_BUCKETS_ODD):
variant = variant_set.choose_variant(bucket)
variant_counts[variant] += 1
self.assertEqual(len(variant_counts), 3)
self.assertEqual(variant_counts["variant_1"], 518)
self.assertEqual(variant_counts["variant_2"], 518)
self.assertEqual(variant_counts[None], 1)
def test_validation_fails(self):
variant_set_cfg_none = None
variant_set_cfg_0 = []
variant_set_cfg_1 = [{"name": "variant_1", "size": 0.25}]
variant_set_cfg_too_big = [
{
"name": "variant_1",
"size": 0.75
},
{
"name": "variant_2",
"size": 0.75
},
]
with self.assertRaises(ValueError):
SingleVariantSet(variant_set_cfg_none)
with self.assertRaises(ValueError):
SingleVariantSet(variant_set_cfg_0)
with self.assertRaises(ValueError):
SingleVariantSet(variant_set_cfg_1)
with self.assertRaises(ValueError):
SingleVariantSet(variant_set_cfg_too_big)
def create_single_variant_set():
cfg = generate_variant_config()
return SingleVariantSet(variants=cfg, num_buckets=NUM_BUCKETS_DEFAULT)