def test_batch_1(): control = ["ABCDE", "FGHIJ", "KLM"] iterable = "abcdefghijklm" def action(b): return "".join(b).upper() result = batch(iterable, 5, action) assert result == control
def test_batch_2(): control = [[ord("A"), ord("B")], [ord("C"), ord("D")]] iterable = "abcd" def prepare_element(elm): return elm.upper() def action(b): return [ord(x) for x in b] result = batch(iterable, 2, action, prepare_element) assert result == control