restore creator_test.py

gangliao-patch-1
gongweibao 8 years ago
parent b359d5c5cd
commit 90c909ac7c

@ -36,14 +36,5 @@ class TestTextFile(unittest.TestCase):
self.assertEqual(e, str(idx * 2) + " " + str(idx * 2 + 1))
class TestRecordIO(unittest.TestCase):
def test_RecordIO(self):
path = os.path.join(
os.path.dirname(__file__), "test_recordio_creator.dat")
reader = paddle.v2.reader.creator.RecordIO(path)
for idx, r in enumerate(reader()):
self.assertSequenceEqual(r, str(idx))
if __name__ == '__main__':
unittest.main()

Loading…
Cancel
Save