mmocr/tests/test_utils/test_parsers.py

# Copyright (c) OpenMMLab. All rights reserved.
import json
from unittest import TestCase

from mmocr.utils import LineJsonParser, LineStrParser


class TestParser(TestCase):

    def test_line_json_parser(self):
        parser = LineJsonParser()
        line = json.dumps(dict(filename='test.jpg', text='mmocr'))
        data = parser(line)
        self.assertEqual(data['filename'], 'test.jpg')
        self.assertEqual(data['text'], 'mmocr')

    def test_line_str_parser(self):
        parser = LineStrParser()
        line = 'test.jpg mmocr'
        data = parser(line)
        self.assertEqual(data['filename'], 'test.jpg')
        self.assertEqual(data['text'], 'mmocr')

        # warnings
        line = 'test test test'
        msg = 'More than two blank spaces were detected. '
        msg += 'Please use LineJsonParser to handle '
        msg += 'annotations with blanks. '
        msg += 'Check Doc '
        msg += 'https://mmocr.readthedocs.io/en/latest/'
        msg += 'tutorials/blank_recog.html '
        msg += 'for details.'
        data = parser(line)
        self.assertWarnsRegex(UserWarning, msg)