Added unit test for batch boundaries