def test_min_ways_respected(self):
for _ in range(10):
num_ways = sampling.sample_num_ways_uniformly(
10,
min_ways=test_utils.MIN_WAYS,
max_ways=test_utils.MAX_WAYS_UPPER_BOUND)
self.assertGreaterEqual(num_ways, test_utils.MIN_WAYS)
def test_max_ways_upper_bound_respected(self):
num_classes = 2 * test_utils.MAX_WAYS_UPPER_BOUND
for _ in range(10):
num_ways = sampling.sample_num_ways_uniformly(
num_classes,
min_ways=test_utils.MIN_WAYS,
max_ways=test_utils.MAX_WAYS_UPPER_BOUND)
self.assertLessEqual(num_ways, test_utils.MAX_WAYS_UPPER_BOUND)
def test_num_classes_respected(self):
num_classes = 10
for _ in range(10):
num_ways = sampling.sample_num_ways_uniformly(
num_classes, min_ways=MIN_WAYS, max_ways=MAX_WAYS_UPPER_BOUND)
self.assertLessEqual(num_ways, num_classes)
