Add tests for wrap.