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)