Add driver tests