mirror of https://github.com/open-mmlab/mmocr.git
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
# 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)
|