mirror of https://github.com/open-mmlab/mmcv.git
92 lines
2.9 KiB
Python
92 lines
2.9 KiB
Python
# Copyright (c) Open-MMLab. All rights reserved.
|
|
import os.path as osp
|
|
|
|
import pytest
|
|
|
|
from mmcv import Config, FileNotFoundError
|
|
|
|
|
|
def test_construct():
|
|
cfg = Config()
|
|
assert cfg.filename is None
|
|
assert cfg.text == ''
|
|
assert len(cfg) == 0
|
|
assert cfg._cfg_dict == {}
|
|
|
|
with pytest.raises(TypeError):
|
|
Config([0, 1])
|
|
|
|
|
|
def test_fromfile():
|
|
for filename in ['a.py', 'b.json', 'c.yaml']:
|
|
cfg_file = osp.join(osp.dirname(__file__), 'data/config', filename)
|
|
cfg = Config.fromfile(cfg_file)
|
|
assert isinstance(cfg, Config)
|
|
assert cfg.filename == cfg_file
|
|
assert cfg.text == open(cfg_file, 'r').read()
|
|
|
|
with pytest.raises(FileNotFoundError):
|
|
Config.fromfile('no_such_file.py')
|
|
with pytest.raises(ValueError):
|
|
Config.fromfile(osp.join(osp.dirname(__file__), 'data/config/a.b.py'))
|
|
with pytest.raises(IOError):
|
|
Config.fromfile(osp.join(osp.dirname(__file__), 'data/color.jpg'))
|
|
|
|
|
|
def test_dict():
|
|
cfg_dict = dict(item1=[1, 2], item2=dict(a=0), item3=True, item4='test')
|
|
|
|
for filename in ['a.py', 'b.json', 'c.yaml']:
|
|
cfg_file = osp.join(osp.dirname(__file__), 'data/config', filename)
|
|
cfg = Config.fromfile(cfg_file)
|
|
|
|
# len(cfg)
|
|
assert len(cfg) == 4
|
|
# cfg.keys()
|
|
assert set(cfg.keys()) == set(cfg_dict.keys())
|
|
assert set(cfg._cfg_dict.keys()) == set(cfg_dict.keys())
|
|
# cfg.values()
|
|
for value in cfg.values():
|
|
assert value in cfg_dict.values()
|
|
# cfg.items()
|
|
for name, value in cfg.items():
|
|
assert name in cfg_dict
|
|
assert value in cfg_dict.values()
|
|
# cfg.field
|
|
assert cfg.item1 == cfg_dict['item1']
|
|
assert cfg.item2 == cfg_dict['item2']
|
|
assert cfg.item2.a == 0
|
|
assert cfg.item3 == cfg_dict['item3']
|
|
assert cfg.item4 == cfg_dict['item4']
|
|
with pytest.raises(AttributeError):
|
|
cfg.not_exist
|
|
# field in cfg, cfg[field], cfg.get()
|
|
for name in ['item1', 'item2', 'item3', 'item4']:
|
|
assert name in cfg
|
|
assert cfg[name] == cfg_dict[name]
|
|
assert cfg.get(name) == cfg_dict[name]
|
|
assert cfg.get('not_exist') is None
|
|
assert cfg.get('not_exist', 0) == 0
|
|
with pytest.raises(KeyError):
|
|
cfg['not_exist']
|
|
assert 'item1' in cfg
|
|
assert 'not_exist' not in cfg
|
|
# cfg.update()
|
|
cfg.update(dict(item1=0))
|
|
assert cfg.item1 == 0
|
|
cfg.update(dict(item2=dict(a=1)))
|
|
assert cfg.item2.a == 1
|
|
|
|
|
|
def test_setattr():
|
|
cfg = Config()
|
|
cfg.item1 = [1, 2]
|
|
cfg.item2 = {'a': 0}
|
|
cfg['item5'] = {'a': {'b': None}}
|
|
assert cfg._cfg_dict['item1'] == [1, 2]
|
|
assert cfg.item1 == [1, 2]
|
|
assert cfg._cfg_dict['item2'] == {'a': 0}
|
|
assert cfg.item2.a == 0
|
|
assert cfg._cfg_dict['item5'] == {'a': {'b': None}}
|
|
assert cfg.item5.a.b is None
|