Write unit tests

[jamesplease]

:see_no_evil:

I came looking to see if there was inspiration here for writing tests for drivers, and saw that there were none! I can tell by the commit history that this is a known TODO, but I figured I'd raise the issue so that others might see it and take a stab at it (maybe I will if I've got time!).