Add a small test suite