Some tests in a torch.test style.