Moved unit tests to separate files to clean up driver