add test for batch