Add tests.
2 files changed