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_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
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
