Add basic tests.

This closes #2.