19 lines
461 B
Python
19 lines
461 B
Python
import os.path as osp
|
|
|
|
import pytest
|
|
|
|
from mmselfsup.datasets.data_sources import ImageNet
|
|
|
|
|
|
def test_imagenet():
|
|
data_source = dict(data_prefix=osp.join(osp.dirname(__file__), '../../'))
|
|
|
|
dataset = ImageNet(**data_source)
|
|
assert len(dataset) == 2
|
|
|
|
with pytest.raises(TypeError):
|
|
dataset = ImageNet(ann_file=1, **data_source)
|
|
|
|
with pytest.raises(RuntimeError):
|
|
dataset = ImageNet(data_prefix=osp.join(osp.dirname(__file__)))
|