# Copyright (c) OpenMMLab. All rights reserved.
import os.path as osp

import pytest

from mmengine import Config, DefaultScope
from mmengine.hub import get_config, get_model
from mmengine.utils import get_installed_path, is_installed

data_path = osp.join(osp.dirname(osp.dirname(__file__)), 'data/')


# mmdet has a more typical config structure, while mmpose has a complex
# config structure
@pytest.mark.skipif(
    not (is_installed('mmdet') and is_installed('mmpose')),
    reason='mmdet and mmpose should be installed')
def test_get_config():
    # Test load base config.
    base_cfg = get_config('mmdet::_base_/models/faster-rcnn_r50_fpn.py')
    package_path = get_installed_path('mmdet')
    test_base_cfg = Config.fromfile(
        osp.join(package_path, '.mim',
                 'configs/_base_/models/faster-rcnn_r50_fpn.py'))
    assert test_base_cfg._cfg_dict == base_cfg._cfg_dict

    # Test load faster_rcnn config
    cfg = get_config('mmdet::faster_rcnn/faster-rcnn_r50_fpn_1x_coco.py')
    test_cfg = Config.fromfile(
        osp.join(package_path, '.mim',
                 'configs/faster_rcnn/faster-rcnn_r50_fpn_1x_coco.py'))
    assert cfg._cfg_dict == test_cfg._cfg_dict

    # Test pretrained
    cfg = get_config(
        'mmdet::faster_rcnn/faster-rcnn_r50_fpn_1x_coco.py', pretrained=True)
    assert cfg.model_path == 'https://download.openmmlab.com/mmdetection/v2.0/faster_rcnn/faster_rcnn_r50_fpn_1x_coco/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth'  # noqa E301

    # Test load mmpose
    get_config(
        'mmpose::face_2d_keypoint/topdown_heatmap/wflw/td-hm_hrnetv2-w18_8xb64-60e_wflw-256x256.py'  # noqa E501
    )


@pytest.mark.skipif(
    not is_installed('mmdet'), reason='mmdet and mmpose should be installed')
def test_get_model():
    # TODO compatible with downstream codebase.
    DefaultScope.get_instance('test_get_model', scope_name='test_scope')
    get_model('mmdet::faster_rcnn/faster-rcnn_r50_fpn_1x_coco.py')
    assert DefaultScope.get_current_instance().scope_name == 'test_scope'
    DefaultScope._instance_dict.pop('test_get_model')