mirror of
https://github.com/open-mmlab/mmfewshot.git
synced 2025-06-03 14:49:43 +08:00
17 lines
399 B
Python
17 lines
399 B
Python
|
import pytest
|
||
|
|
from mmcv import ConfigDict
|
||
|
|
|
||
|
|
from mmfewshot.utils.check_config import check_config
|
||
|
|
|
||
|
|
|
||
|
|
def test_check_config():
|
||
|
|
config = dict(task_type='mmdet')
|
||
|
|
cfg = ConfigDict(config)
|
||
|
|
check_config(cfg)
|
||
|
|
with pytest.raises(AttributeError):
|
||
|
|
cfg.pop('task_type')
|
||
|
|
check_config(cfg)
|
||
|
|
with pytest.raises(ValueError):
|
||
|
|
cfg.task_type = 'cls'
|
||
|
|
check_config(cfg)
|