Add imdb_test.py

8 years ago · 349319e5ed
parent 89d300191e
commit 349319e5ed
1 changed files with 7 additions and 1 deletions
--- a/python/paddle/v2/dataset/tests/imdb_test.py
+++ b/python/paddle/v2/dataset/tests/imdb_test.py
@ -1,6 +1,12 @@
 import paddle.v2.dataset.common
+import tarfile

 URL = 'http://ai.stanford.edu/%7Eamaas/data/sentiment/aclImdb_v1.tar.gz'
 MD5 = '7c2ac02c03563afcf9b574c7e56c153a'

-paddle.v2.dataset.common.download
+tarf = tarfile.open(paddle.v2.dataset.common.download(URL, 'imdb', MD5))
+
+tf = tarf.next()
+while tf != None:
+    print tf.name
+    tf = tarf.next()