mirror of
https://github.com/open-mmlab/mmcv.git
synced 2025-06-03 21:54:52 +08:00
979a355d2a
* add windows CI * clean versions * only allow pt1.7 on windows * fix windows install issue * add win cpu * fix win command * clean unnecessary command * resolve turbojpeg & tempfile on win * replace os.readlink with os.path.realpath * fix windows ci * close file before removing it * fix windows ci * fix symlink on windows * fix windows ci * fix windows ci * fix windows ci * fix windows ci * fix windows ci * fix windows ci * fix windows ci * fix windows ci * fix windows ci * modify according to comment Co-authored-by: zhouzaida <zhouzaida@163.com>
61 lines
2.1 KiB
Python
61 lines
2.1 KiB
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
import os.path as osp
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
import mmcv
|
|
|
|
|
|
def test_is_filepath():
|
|
assert mmcv.is_filepath(__file__)
|
|
assert mmcv.is_filepath('abc')
|
|
assert mmcv.is_filepath(Path('/etc'))
|
|
assert not mmcv.is_filepath(0)
|
|
|
|
|
|
def test_fopen():
|
|
assert hasattr(mmcv.fopen(__file__), 'read')
|
|
assert hasattr(mmcv.fopen(Path(__file__)), 'read')
|
|
|
|
|
|
def test_check_file_exist():
|
|
mmcv.check_file_exist(__file__)
|
|
with pytest.raises(FileNotFoundError):
|
|
mmcv.check_file_exist('no_such_file.txt')
|
|
|
|
|
|
def test_scandir():
|
|
folder = osp.join(osp.dirname(osp.dirname(__file__)), 'data/for_scan')
|
|
filenames = ['a.bin', '1.txt', '2.txt', '1.json', '2.json']
|
|
assert set(mmcv.scandir(folder)) == set(filenames)
|
|
assert set(mmcv.scandir(Path(folder))) == set(filenames)
|
|
assert set(mmcv.scandir(folder, '.txt')) == set(
|
|
[filename for filename in filenames if filename.endswith('.txt')])
|
|
assert set(mmcv.scandir(folder, ('.json', '.txt'))) == set([
|
|
filename for filename in filenames
|
|
if filename.endswith(('.txt', '.json'))
|
|
])
|
|
assert set(mmcv.scandir(folder, '.png')) == set()
|
|
|
|
# path of sep is `\\` in windows but `/` in linux, so osp.join should be
|
|
# used to join string for compatibility
|
|
filenames_recursive = [
|
|
'a.bin', '1.txt', '2.txt', '1.json', '2.json',
|
|
osp.join('sub', '1.json'),
|
|
osp.join('sub', '1.txt'), '.file'
|
|
]
|
|
# .file starts with '.' and is a file so it will not be scanned
|
|
assert set(mmcv.scandir(folder, recursive=True)) == set(
|
|
[filename for filename in filenames_recursive if filename != '.file'])
|
|
assert set(mmcv.scandir(Path(folder), recursive=True)) == set(
|
|
[filename for filename in filenames_recursive if filename != '.file'])
|
|
assert set(mmcv.scandir(folder, '.txt', recursive=True)) == set([
|
|
filename for filename in filenames_recursive
|
|
if filename.endswith('.txt')
|
|
])
|
|
with pytest.raises(TypeError):
|
|
list(mmcv.scandir(123))
|
|
with pytest.raises(TypeError):
|
|
list(mmcv.scandir(folder, 111))
|