Add tests

No gem is a real gem without tests! :metal: