diff --git a/python/paddle/v2/dataset/tests/imdb_test.py b/python/paddle/v2/dataset/tests/imdb_test.py index fafaf414044a4a782fe84d5840dc2402f76a56cf..5cfb6203e78ec14f2cc5474876a6ed7de7d491e9 100644 --- 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()