Wrap test code