diff --git a/python/paddle/v2/framework/tests/test_exp_op.py b/python/paddle/v2/framework/tests/test_exp_op.py
index 5a004f6fe2f4213cd64dfcb1438522365b8a37a6..0ec41e56a03fa03ec23b8caa5b6fd00d250647ed 100644
--- a/python/paddle/v2/framework/tests/test_exp_op.py
+++ b/python/paddle/v2/framework/tests/test_exp_op.py
@@ -15,8 +15,8 @@ class TestExp(OpTest):
         self.check_output()
 
     def test_check_grad(self):
-        self.check_grad(["X"], "Y", max_relative_error=0.007)
+        self.check_grad(['X'], 'Y', max_relative_error=0.007)
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     unittest.main()
diff --git a/python/paddle/v2/framework/tests/test_relu_op.py b/python/paddle/v2/framework/tests/test_relu_op.py
index 58a0872db4185784feaffaa5dccbde9934dbb21e..c9af0c2ba7e0d99c83898bfc6744ecf89cbfc1c8 100644
--- a/python/paddle/v2/framework/tests/test_relu_op.py
+++ b/python/paddle/v2/framework/tests/test_relu_op.py
@@ -6,17 +6,15 @@ from op_test import OpTest
 class TestRelu(OpTest):
     def setUp(self):
         self.op_type = "relu"
-        self.inputs = {
-            'X': np.random.uniform(-1, 1, [11, 17]).astype("float32")
-        }
+        self.inputs = {'X': np.random.uniform(-1, 1, [4, 4]).astype("float32")}
         self.outputs = {'Y': np.maximum(self.inputs['X'], 0)}
 
     def test_check_output(self):
         self.check_output()
 
     def test_check_grad(self):
-        self.check_grad(["X"], "Y", max_relative_error=0.007)
+        self.check_grad(['X'], 'Y', max_relative_error=0.007)
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     unittest.main()
diff --git a/python/paddle/v2/framework/tests/test_sigmoid_op.py b/python/paddle/v2/framework/tests/test_sigmoid_op.py
index 2316e49eff7bb1cdb53acb3889a6ef05060b59f3..cf05e934d5f83544dc8fa78d6b3b54a2a38d55b3 100644
--- a/python/paddle/v2/framework/tests/test_sigmoid_op.py
+++ b/python/paddle/v2/framework/tests/test_sigmoid_op.py
@@ -15,8 +15,8 @@ class TestSigmoid(OpTest):
         self.check_output()
 
     def test_check_grad(self):
-        self.check_grad(["X"], "Y", max_relative_error=0.007)
+        self.check_grad(['X'], 'Y', max_relative_error=0.007)
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     unittest.main()