Add base test for the batch functionality