Add tests for sanity-checking