Add unit testing

Signed-off-by: Casey Dahlin <casey.dahlin@gmail.com>