Move tests to the driver module.