got rid of duplicate batch_size assignment in testcases