Add more tests for wrap.