add test

huggingface · Apr 30, 2024 · be4e44b · be4e44b
1 parent 82c1566
commit be4e44b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/models/vitmatte/test_image_processing_vitmatte.py b/tests/models/vitmatte/test_image_processing_vitmatte.py
@@ -192,3 +192,7 @@ def test_padding(self):
         image = np.random.randn(3, 249, 491)
         images = image_processing.pad_image(image)
         assert images.shape == (3, 256, 512)
+
+        image = np.random.randn(3, 249, 512)
+        images = image_processing.pad_image(image)
+        assert images.shape == (3, 256, 512)