Move wrapper tests to torch.test()