Fix resize not using order.

Project-MONAI · Mar 5, 2020 · abbb743 · abbb743
1 parent aa8dfbf
commit abbb743
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/monai/transforms/transforms.py b/monai/transforms/transforms.py
@@ -115,7 +115,7 @@ def __init__(self, output_shape, order=1, mode='reflect', cval=0,
         self.anti_aliasing_sigma = anti_aliasing_sigma
 
     def __call__(self, img):
-        return resize(img, self.output_shape,
+        return resize(img, self.output_shape, order=self.order,
                       mode=self.mode, cval=self.cval,
                       clip=self.clip, preserve_range=self.preserve_range,
                       anti_aliasing=self.anti_aliasing, 

diff --git a/tests/test_resize.py b/tests/test_resize.py
@@ -30,8 +30,9 @@ def test_invalid_inputs(self, _, order, raises):
             resize(self.imt)
 
     @parameterized.expand([
-        ((64, 64, 3), 1, 'reflect', 0, True, True, True, None),
-        ((32, 32, 3), 2, 'constant', 1, False, False, False, None)
+        ((1, 1, 64, 64), 1, 'reflect', 0, True, True, True, None),
+        ((1, 1, 32, 32), 2, 'constant', 3, False, False, False, None),
+        ((1, 1, 256, 256), 3, 'constant', 3, False, False, False, None),
     ])
     def test_correct_results(self, output_shape, order, mode, 
                              cval, clip, preserve_range,