Add test for batches count calculation.
Added test to ensure all batches are included.