def test_buckets_correct_size(self):
config = {"count": 2}
count = 0
kernel = {k: False for k in range(9)}
values = Solution(0.0, ((k, randbelow(100)) for k in range(9)))
sizes = [6, 3]
for bucket, length in zip(
decresing_size_bucket(kernel, values, bucket_sort, {}, **config), sizes
):
count += 1
self.assertEqual(len(bucket), length)
self.assertEqual(count, 2)
def _build_solution(self, bucket):
gen = ((letter, int(index < 4)) for index, letter in enumerate(bucket))
return Solution(12, gen)
def build_kernel_fixed_size():
kernel = {k: i % 2 == 0 for i, k in enumerate(ascii_letters)}
values = Solution(0, ((k, randbelow(100)) for k in ascii_letters))
return kernel, values
def build_kernel():
kernel = {k: i % 2 == 0 for i, k in enumerate(ascii_lowercase)}
values = Solution(0.0, ((k, -i) for i, k in enumerate(ascii_lowercase)))
return kernel, values