Adding tests and some docs comments for batch predictions with test split