More robust assertion checks in test_to_tensor_video (#4258)

db530d8c · Nicolas Hug · GitHub · d7fa36f2 · db530d8c
Unverified Commit db530d8c authored Aug 16, 2021 by Nicolas Hug Committed by GitHub Aug 16, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

test/test_transforms_video.py test/test_transforms_video.py +4 -0

No files found.
--- a/test/test_transforms_video.py
+++ b/test/test_transforms_video.py
@@ -133,12 +133,16 @@ class TestVideoTransforms():
        with pytest.raises(TypeError):
            np_rng = np.random.RandomState(0)
            trans(np_rng.rand(numFrames, height, width, 1).tolist())
+        with pytest.raises(TypeError):
            trans(torch.rand((numFrames, height, width, 1), dtype=torch.float))

        with pytest.raises(ValueError):
            trans(torch.ones((3, numFrames, height, width, 3), dtype=torch.uint8))
+        with pytest.raises(ValueError):
            trans(torch.ones((height, width, 3), dtype=torch.uint8))
+        with pytest.raises(ValueError):
            trans(torch.ones((width, 3), dtype=torch.uint8))
+        with pytest.raises(ValueError):
            trans(torch.ones((3), dtype=torch.uint8))

        trans.__repr__()