Feature/add batch unit test

原来那个单有冲突，这是新开的pr