Moved batch_test.py to new driver