mirrors/mmclassification
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmclassification.git synced 2025-06-03 21:53:55 +08:00
Code Issues Projects Releases Wiki Activity

fix test_dataset

Browse Source
This commit is contained in:
yanglei 2020-07-06 00:43:22 +08:00 committed by chenkai
parent a33b7aa91d
commit f9b1f18f28
1 changed files with 1 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/test_dataset.py
View File

@ -11,8 +11,7 @@ import pytest
from mmcls.datasets import (DATASETS, BaseDataset, ClassBalancedDataset, from mmcls.datasets import (DATASETS, BaseDataset, ClassBalancedDataset,
ConcatDataset, RepeatDataset) ConcatDataset, RepeatDataset)
from mmcls.datasets.utils import (check_integrity, from mmcls.datasets.utils import check_integrity, rm_suffix
download_and_extract_archive, rm_suffix)
@pytest.mark.parametrize( @pytest.mark.parametrize(
@ -113,10 +112,3 @@ def test_dataset_utils():
tmp_file = tempfile.NamedTemporaryFile() tmp_file = tempfile.NamedTemporaryFile()
assert check_integrity(tmp_file.name, md5=None) assert check_integrity(tmp_file.name, md5=None)
assert not check_integrity(tmp_file.name, md5=2333) assert not check_integrity(tmp_file.name, md5=2333)
# test download_and_extract_archive
url = 'http://yann.lecun.com/exdb/mnist/train-labels-idx1-ubyte.gz'
md5 = 'd53e105ee54ea40749a09fcbcd1e9432'
tmp_dir = tempfile.TemporaryDirectory()
download_and_extract_archive(
url, download_root=tmp_dir.name, md5=md5, remove_finished=True)