update test for fake reader decorator

revert-13637-optimize-opyreader
qiaolongfei 6 years ago
parent bda05dc387
commit 4fc5991728

@ -211,6 +211,7 @@ class TestFakeReader(unittest.TestCase):
data_num = 100 data_num = 100
fake_reader = paddle.reader.fake(reader, data_num) fake_reader = paddle.reader.fake(reader, data_num)
for _ in range(10):
i = 0 i = 0
for data in fake_reader(): for data in fake_reader():
self.assertEqual(data, 0) self.assertEqual(data, 0)

Loading…
Cancel
Save