Method test_transpose

tests/transforms/test_transpose.py:31–37

(self, im, indices)

class TestTranspose(unittest.TestCase):
    @parameterized.expand(TESTS)
    def test_transpose(self, im, indices):
        tr = Transpose(indices)
        out1 = tr(im)
        if isinstance(im, torch.Tensor):
            im = im.cpu().numpy()
        out2 = np.transpose(im, indices)
        assert_allclose(out1, out2, type_test=False)


if __name__ == "__main__":