Skip to content

Commit 8c1eb86

Browse files
committed
update unittest
1 parent 0dacbbe commit 8c1eb86

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

python/paddle/fluid/tests/unittests/test_multiple_reader.py renamed to python/paddle/fluid/tests/unittests/test_multi_file_reader.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -61,8 +61,12 @@ def main(self, thread_num):
6161
exe.run(fluid.default_startup_program())
6262

6363
batch_count = 0
64-
while not data_files.eof():
65-
img_val, = exe.run(fetch_list=[img])
64+
while True:
65+
try:
66+
img_val, = exe.run(fetch_list=[img])
67+
except fluid.core.EnforceNotMet as ex:
68+
self.assertIn("There is no next data.", ex.message)
69+
break
6670
batch_count += 1
6771
self.assertLessEqual(img_val.shape[0], self.batch_size)
6872
data_files.reset()

0 commit comments

Comments
 (0)