Add tests

Nothing different about the API. Just some basic tests.