Add tests to gem