From 50b8ec0564534b07fe5fdacd91db6a21fafd0559 Mon Sep 17 00:00:00 2001
From: chengduoZH <zhaochengduo@163.com>
Date: Thu, 21 Sep 2017 19:21:12 +0800
Subject: [PATCH] fix unit test

---
 .../v2/framework/tests/test_pool2d_op.py      | 22 ++++++++++---------
 .../v2/framework/tests/test_pool3d_op.py      | 22 ++++++++++---------
 2 files changed, 24 insertions(+), 20 deletions(-)

diff --git a/python/paddle/v2/framework/tests/test_pool2d_op.py b/python/paddle/v2/framework/tests/test_pool2d_op.py
index 549898ec94e..cf327508f71 100644
--- a/python/paddle/v2/framework/tests/test_pool2d_op.py
+++ b/python/paddle/v2/framework/tests/test_pool2d_op.py
@@ -62,7 +62,8 @@ class TestPool2d_Op(OpTest):
         self.check_output()
 
     def test_check_grad(self):
-        self.check_grad(set(['Input']), 'Output', max_relative_error=0.07)
+        if self.pool_type != "max":
+            self.check_grad(set(['Input']), 'Output', max_relative_error=0.07)
 
     def initTestCase(self):
         self.pool_type = "ave"
@@ -84,15 +85,16 @@ class TestCase2(TestPool2d_Op):
         self.paddings = [1, 1]
 
 
-# class TestCase1(TestPool2d_Op):
-#     def initTestCase(self):
-#         self.op_type = "pool2d"
-#         self.pool_type = "max"
-#         self.pool2D_forward_naive = max_pool2D_forward_naive
-#         self.shape = [2, 3, 5, 5]
-#         self.ksize = [3, 3]
-#         self.strides = [1, 1]
-#         self.paddings = [1, 1]
+class TestCase1(TestPool2d_Op):
+    def initTestCase(self):
+        self.op_type = "pool2d"
+        self.pool_type = "max"
+        self.pool2D_forward_naive = max_pool2D_forward_naive
+        self.shape = [2, 3, 5, 5]
+        self.ksize = [3, 3]
+        self.strides = [1, 1]
+        self.paddings = [1, 1]
+
 
 if __name__ == '__main__':
     unittest.main()
diff --git a/python/paddle/v2/framework/tests/test_pool3d_op.py b/python/paddle/v2/framework/tests/test_pool3d_op.py
index f8e9a768e03..cfd0ced150b 100644
--- a/python/paddle/v2/framework/tests/test_pool3d_op.py
+++ b/python/paddle/v2/framework/tests/test_pool3d_op.py
@@ -72,7 +72,8 @@ class TestPool3d_Op(OpTest):
         self.check_output()
 
     def test_check_grad(self):
-        self.check_grad(set(['Input']), 'Output', max_relative_error=0.07)
+        if self.pool_type != "max":
+            self.check_grad(set(['Input']), 'Output', max_relative_error=0.07)
 
     def initTestCase(self):
         self.pool_type = "ave"
@@ -94,15 +95,16 @@ class TestCase1(TestPool3d_Op):
         self.paddings = [1, 1, 1]
 
 
-# class TestCase2(TestPool3d_Op):
-#     def initTestCase(self):
-#         self.op_type = "pool3d"
-#         self.pool_type = "max"
-#         self.pool3D_forward_naive = max_pool3D_forward_naive
-#         self.shape = [2, 3, 5, 5, 5]
-#         self.ksize = [3, 3, 3]
-#         self.strides = [1, 1, 1]
-#         self.paddings = [1, 1, 1]
+class TestCase2(TestPool3d_Op):
+    def initTestCase(self):
+        self.op_type = "pool3d"
+        self.pool_type = "max"
+        self.pool3D_forward_naive = max_pool3D_forward_naive
+        self.shape = [2, 3, 5, 5, 5]
+        self.ksize = [3, 3, 3]
+        self.strides = [1, 1, 1]
+        self.paddings = [1, 1, 1]
+
 
 if __name__ == '__main__':
     unittest.main()
-- 
GitLab