Add tests for wrap()