Update test_image_filter.py

ASEM000 · Sep 12, 2023 · 70602ed · 70602ed
1 parent 4a2a441
commit 70602ed
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/test_image_filter.py b/tests/test_image_filter.py
@@ -429,3 +429,10 @@ def test_laplacian():
         sk.image.Filter2D(kernel)(x),
         atol=1e-5,
     )
+
+
+def test_center_crop():
+    x = jnp.arange(1, 26).reshape(1, 5, 5)
+    assert sk.image.CenterCrop2D(3)(x).shape == (1, 3, 3)
+    assert sk.image.CenterCrop2D(0)(x).shape == (1, 0, 0)
+    npt.assert_allclose(sk.image.CenterCrop2D(3)(x), x[:, 1:-1, 1:-1])