improved mem. tests for AutoBatch