optimize unit test of test_split_ids_op

7 years ago · f4c4c6179c
parent 147bf00ffe
commit f4c4c6179c
1 changed files with 12 additions and 11 deletions
--- a/python/paddle/fluid/tests/unittests/test_split_ids_op.py
+++ b/python/paddle/fluid/tests/unittests/test_split_ids_op.py
@ -69,6 +69,7 @@ class TestSpliteIds(unittest.TestCase):

        op = Operator("split_ids", Ids="X", Out=outs_name)

+        for _ in range(3):
            op.run(scope, place)

            for i in range(len(outs)):