Add tests for wrap method.