diff --git a/python/paddle/fluid/tests/unittests/test_similarity_focus_op.py b/python/paddle/fluid/tests/unittests/test_similarity_focus_op.py
index bd3b2782aea46598858920035f58e1937efded55..b3833f05f1aa3aac7b5bcc5b6fdc138870cc8844 100755
--- a/python/paddle/fluid/tests/unittests/test_similarity_focus_op.py
+++ b/python/paddle/fluid/tests/unittests/test_similarity_focus_op.py
@@ -57,8 +57,7 @@ class TestSimilarityFocusOp(OpTest):
                         if cnt == min(y_dim, z_dim):
                             break
                     channel[index] = -1
-            res = res.reshape(1, y_dim, z_dim)
-            res = res.repeat([x_dim], axis=0)
+            res = res.reshape(1, y_dim, z_dim).repeat([x_dim], axis=0)
             res = res.reshape(1, x_dim, y_dim, z_dim)
             if output is not None:
                 output = np.concatenate((output, res), axis=0)