add unitest for param out of range

Author: LDOUBLEV

python/paddle/fluid/tests/unittests/test_pool1d_api.py

@@ -340,6 +340,36 @@ def run7():
 
         self.assertRaises(ValueError, run7)
 
+        def run_kernel_out_of_range():
+            with fluid.dygraph.guard():
+                input_np = np.random.uniform(-1, 1,
+                                             [2, 3, 32]).astype(np.float32)
+                input_pd = fluid.dygraph.to_variable(input_np)
+                padding = 0
+                res_pd = F.avg_pool1d(
+                    input_pd,
+                    kernel_size=-1,
+                    stride=2,
+                    padding=padding,
+                    ceil_mode=True)
+
+        self.assertRaises(ValueError, run_kernel_out_of_range)
+
+        def run_stride_out_of_range():
+            with fluid.dygraph.guard():
+                input_np = np.random.uniform(-1, 1,
+                                             [2, 3, 32]).astype(np.float32)
+                input_pd = fluid.dygraph.to_variable(input_np)
+                padding = 0
+                res_pd = F.avg_pool1d(
+                    input_pd,
+                    kernel_size=2,
+                    stride=0,
+                    padding=padding,
+                    ceil_mode=True)
+
+        self.assertRaises(ValueError, run_stride_out_of_range)
+
 
 if __name__ == '__main__':
     unittest.main()

python/paddle/fluid/tests/unittests/test_pool2d_api.py

@@ -494,6 +494,36 @@ def run9():
 
         self.assertRaises(ValueError, run9)
 
+        def run_kernel_out_of_range():
+            with fluid.dygraph.guard():
+                input_np = np.random.uniform(-1, 1,
+                                             [2, 3, 32, 32]).astype(np.float32)
+                input_pd = fluid.dygraph.to_variable(input_np)
+                res_pd = avg_pool2d(
+                    input_pd,
+                    kernel_size=[-1, 2],
+                    stride=2,
+                    padding=0,
+                    ceil_mode=False,
+                    data_format='NHWC')
+
+        self.assertRaises(ValueError, run_kernel_out_of_range)
+
+        def run_stride_out_of_range():
+            with fluid.dygraph.guard():
+                input_np = np.random.uniform(-1, 1,
+                                             [2, 3, 32, 32]).astype(np.float32)
+                input_pd = fluid.dygraph.to_variable(input_np)
+                res_pd = avg_pool2d(
+                    input_pd,
+                    kernel_size=3,
+                    stride=[0, 2],
+                    padding=0,
+                    ceil_mode=False,
+                    data_format='NHWC')
+
+        self.assertRaises(ValueError, run_stride_out_of_range)
+
 
 if __name__ == '__main__':
     unittest.main()

python/paddle/fluid/tests/unittests/test_pool3d_api.py

@@ -471,6 +471,34 @@ def run10():
 
         self.assertRaises(ValueError, run10)
 
+        def run_kernel_out_of_range():
+            with fluid.dygraph.guard():
+                input_np = np.random.uniform(
+                    -1, 1, [2, 3, 32, 32, 32]).astype(np.float32)
+                input_pd = fluid.dygraph.to_variable(input_np)
+                res_pd = avg_pool3d(
+                    input_pd,
+                    kernel_size=-1,
+                    stride=2,
+                    padding="VALID",
+                    ceil_mode=True)
+
+        self.assertRaises(ValueError, run_kernel_out_of_range)
+
+        def run_size_out_of_range():
+            with fluid.dygraph.guard():
+                input_np = np.random.uniform(
+                    -1, 1, [2, 3, 32, 32, 32]).astype(np.float32)
+                input_pd = fluid.dygraph.to_variable(input_np)
+                res_pd = avg_pool3d(
+                    input_pd,
+                    kernel_size=2,
+                    stride=0,
+                    padding="VALID",
+                    ceil_mode=True)
+
+        self.assertRaises(ValueError, run_size_out_of_range)
+
 
 if __name__ == '__main__':
     unittest.main()