diff --git a/python/paddle/fluid/tests/unittests/test_py_reader_combination.py b/python/paddle/fluid/tests/unittests/test_py_reader_combination.py index 21f5fe4b581ab74bae52d952020ed2183820c471..2d977caa03369840d4ac31344195878a9998f685 100644 --- a/python/paddle/fluid/tests/unittests/test_py_reader_combination.py +++ b/python/paddle/fluid/tests/unittests/test_py_reader_combination.py @@ -43,6 +43,12 @@ class TestPyReaderCombination(unittest.TestCase): self.assertTrue(np.array_equal(image1, image2)) self.assertTrue(np.array_equal(label1, label2)) + # FIXME(zjl): do not know why Python 35 would raise SIGABRT if not reset reader + # manually. + def _reset_iterable_reader(self, py_reader): + if py_reader.iterable: + py_reader._loader._reset() + def main_impl(self, place): with fluid.program_guard(fluid.Program(), fluid.Program()): image = fluid.layers.data( @@ -70,6 +76,9 @@ class TestPyReaderCombination(unittest.TestCase): self.assertEqual(batch_num, max_num) + self._reset_iterable_reader(py_reader1) + self._reset_iterable_reader(py_reader2) + def get_places(self): if fluid.is_compiled_with_cuda(): return [fluid.CUDAPlace(0), fluid.CPUPlace()]