Add basic tests.

Fixes #5.