From dd926498e7e61b25250c8f59d91afe57ab24098a Mon Sep 17 00:00:00 2001
From: Yibing Liu <liuyibing01@baidu.com>
Date: Mon, 11 Sep 2017 20:09:21 -0700
Subject: [PATCH] adapt to the new test framework

---
 .../v2/framework/tests/test_reshape_op.py     | 27 +++++++------------
 1 file changed, 9 insertions(+), 18 deletions(-)

diff --git a/python/paddle/v2/framework/tests/test_reshape_op.py b/python/paddle/v2/framework/tests/test_reshape_op.py
index 50653f58ee1..16bb6bb2af6 100644
--- a/python/paddle/v2/framework/tests/test_reshape_op.py
+++ b/python/paddle/v2/framework/tests/test_reshape_op.py
@@ -1,29 +1,20 @@
 import unittest
 import numpy as np
-from gradient_checker import GradientChecker, Operator
-from op_test_util import OpTestMeta
+from op_test import OpTest
 
 
-class TestReshapeOp(unittest.TestCase):
-    __metaclass__ = OpTestMeta
-
+class TestReshapeOp(OpTest):
     def setUp(self):
-        self.type = "reshape"
-        self.inputs = {'X': np.random.random((37, 51)).astype("float32"), }
-        self.attrs = {'shape': [51 * 37]}
+        self.op_type = "reshape"
+        self.inputs = {'X': np.random.random((10, 20)).astype("float32")}
+        self.attrs = {'shape': [10 * 20]}
         self.outputs = {'Out': self.inputs['X'].reshape(self.attrs['shape'])}
 
+    def test_check_output(self):
+        self.check_output()
 
-class TestReshapeGradOp(GradientChecker):
-    def setUp(self):
-        self.op = Operator("reshape", X='X', Out='Out', shape=[5, 40])
-        self.inputs = {"X": np.random.random((10, 20)).astype("float32")}
-
-    def test_normal(self):
-        self.check_grad(self.op, self.inputs, ["X"], "Out")
-
-    def test_dev_compare(self):
-        self.compare_grad(self.op, self.inputs)
+    def test_check_grad(self):
+        self.check_grad(["X"], "Out")
 
 
 if __name__ == '__main__':
-- 
GitLab