Please don't change tests, unless the output changes. Your modifications add a lot of new things, and there are actually different "sections" to it:

- Testing limit less than batchSize
- Testing limit equal than batchSize

Please create a different test file, it makes a lot easier to manage tests and expected/returned output.