update test for fake reader decorator

6 years ago · 4fc5991728
parent bda05dc387
commit 4fc5991728
1 changed files with 6 additions and 5 deletions
--- a/python/paddle/reader/tests/decorator_test.py
+++ b/python/paddle/reader/tests/decorator_test.py
@ -211,6 +211,7 @@ class TestFakeReader(unittest.TestCase):

        data_num = 100
        fake_reader = paddle.reader.fake(reader, data_num)
+        for _ in range(10):
            i = 0
            for data in fake_reader():
                self.assertEqual(data, 0)