add test when crop size == img size (#291)

d3e78103 · Alykhan Tejani · Francisco Massa · 56b973b5 · d3e78103
Commit d3e78103 authored Oct 13, 2017 by Alykhan Tejani Committed by Francisco Massa Oct 13, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 9 additions and 0 deletions

test/test_transforms.py test/test_transforms.py +9 -0

No files found.
--- a/test/test_transforms.py
+++ b/test/test_transforms.py
@@ -190,6 +190,15 @@ class Tester(unittest.TestCase):
        assert result.size(1) == oheight
        assert result.size(2) == owidth

+        result = transforms.Compose([
+            transforms.ToPILImage(),
+            transforms.RandomCrop((height, width)),
+            transforms.ToTensor()
+        ])(img)
+        assert result.size(1) == height
+        assert result.size(2) == width
+        assert np.allclose(img.numpy(), result.numpy())
+
    def test_pad(self):
        height = random.randint(10, 32) * 2
        width = random.randint(10, 32) * 2