tests/test_core/test_end2end_vis.py

# Copyright (c) OpenMMLab. All rights reserved.
import numpy as np

from mmocr.core import det_recog_show_result


def test_det_recog_show_result():
    img = np.ones((100, 100, 3), dtype=np.uint8) * 255
    det_recog_res = {
        'result': [{
            'box': [51, 88, 51, 62, 85, 62, 85, 88],
            'box_score': 0.9417,
            'text': 'hell',
            'text_score': 0.8834
        }]
    }

    vis_img = det_recog_show_result(img, det_recog_res)

    assert vis_img.shape[0] == 100
    assert vis_img.shape[1] == 200
    assert vis_img.shape[2] == 3

    det_recog_res['result'][0]['text'] = '中文'
    det_recog_show_result(img, det_recog_res)