def test_include_leftovers_causes_remaining_items_to_be_yielded_as_the_final_batch(self): result = list(latr(self.items) >> BatchesOf(3, on_leftovers = include_leftovers)) self.assertEqual(len(result), 3) self.assertEqual(result[0], self.items[0:3]) self.assertEqual(result[1], self.items[3:6]) self.assertEqual(result[2], self.items[6:8])
def test_pad_leftovers_with_causes_a_default_value_to_be_used_in_place_of_missing_items(self):
result = list(latr(self.items) >> BatchesOf(5, on_leftovers = pad_leftovers_with('stuff')))
self.assertEqual(len(result), 2)
self.assertEqual(result[0], self.items[0:5])
self.assertEqual(result[1], self.items[5:8] + ['stuff', 'stuff'])
def test_discard_leftovers_causes_remaining_items_to_be_discarded(self): result = list(latr(self.items) >> BatchesOf(3, on_leftovers = discard_leftovers)) self.assertEqual(len(result), 2) self.assertEqual(result[0], self.items[0:3]) self.assertEqual(result[1], self.items[3:6])
def test_items_that_do_not_fit_in_a_batch_are_discarded_by_default(self): result = list(latr(self.items) >> BatchesOf(3)) self.assertEqual(len(result), 2) self.assertEqual(result[0], self.items[0:3]) self.assertEqual(result[1], self.items[3:6])
def test_all_items_are_returned_when_batch_size_divides_items_evenly(self): result = list(latr(self.items) >> BatchesOf(4)) self.assertEqual(len(result), 2) self.assertEqual(result[0] + result[1], self.items)
def test_1_batch_is_returned_when_size_exactly_equals_number_of_items(self):
result = list(latr(self.items) >> BatchesOf(len(self.items)))
self.assertEqual(len(result), 1)
self.assertEqual(result[0], self.items)
def test_each_list_has_1_item_when_size_is_1(self):
result = latr(self.items) >> BatchesOf(1)
for i in self.items:
self.assertEqual(result.next(), [i])
def test_it_returns_an_Iterator(self): result = latr(self.items) >> BatchesOf(1) self.assertTrue(isinstance(result, Iterator))
