def test_create_batches_creates_at_generator(self, _, create_at_generator):
create_batches(*self.test_args)
create_at_generator.assert_called_once_with(self.at, self.interval)
def test_create_batches_return_tuple_of_batches_and_at_generator(self, batch_items, create_at_generator):
ret = create_batches(*self.test_args)
self.assertEqual(ret, (batch_items.return_value, create_at_generator.return_value))
def test_create_batches_batches_items(self, batch_items, *args):
create_batches(*self.test_args)
batch_items.assert_called_once_with(self.command, self.items, self.parallel)