|
|
@@ -0,0 +1,244 @@
|
|
|
+import unittest
|
|
|
+import base64
|
|
|
+from dataclasses import dataclass
|
|
|
+from pathlib import Path
|
|
|
+
|
|
|
+import requests
|
|
|
+
|
|
|
+# url = 'http://localhost:8080'
|
|
|
+url = 'http://192.168.199.107:48080'
|
|
|
+
|
|
|
+# header = {
|
|
|
+# 'Authorization': 'Bearer 9679c2b3-b90b-4029-a3c7-f347b4d242f7'
|
|
|
+# }
|
|
|
+
|
|
|
+# 带线框识别比较差 例:6.27_02.png
|
|
|
+
|
|
|
+def send_request(image_path, image_type='0'):
|
|
|
+ with open(image_path, 'rb') as f:
|
|
|
+ img_str: str = base64.encodebytes(f.read()).decode('utf-8')
|
|
|
+ r = requests.post(f'{url}/ocr_system/regbook', json={'image': img_str, 'image_type': image_type})
|
|
|
+ print(r.json())
|
|
|
+ return r.json()
|
|
|
+
|
|
|
+
|
|
|
+@dataclass
|
|
|
+class ResultItem0:
|
|
|
+ status: str
|
|
|
+ orientation: int
|
|
|
+ name: str
|
|
|
+ id: str
|
|
|
+ gender: str
|
|
|
+ birthplace: str
|
|
|
+ native_place: str
|
|
|
+ blood_type: str
|
|
|
+ religion: str
|
|
|
+
|
|
|
+@dataclass
|
|
|
+class ResultItem1:
|
|
|
+ status: str
|
|
|
+ orientation: int
|
|
|
+ type: str
|
|
|
+ address: str
|
|
|
+
|
|
|
+
|
|
|
+class TestSchoolCertOcr(unittest.TestCase):
|
|
|
+ def _helper0(self, image_path, item: ResultItem0, image_type='0'):
|
|
|
+ root = Path(__file__).parent
|
|
|
+ image_path = str(root / image_path)
|
|
|
+ r = send_request(image_path, image_type)
|
|
|
+ self.assertEqual(item, ResultItem0(status=r['status'],
|
|
|
+ orientation=r['result']['orientation'],
|
|
|
+ name=r['result']['name']['text'],
|
|
|
+ id=r['result']['id']['text'],
|
|
|
+ gender=r['result']['gender']['text'],
|
|
|
+ birthplace=r['result']['birthplace']['text'],
|
|
|
+ native_place=r['result']['native_place']['text'],
|
|
|
+ blood_type=r['result']['blood_type']['text'],
|
|
|
+ religion=r['result']['religion']['text']
|
|
|
+ ))
|
|
|
+
|
|
|
+ def _helper1(self, image_path, item: ResultItem1, image_type='1'):
|
|
|
+ root = Path(__file__).parent
|
|
|
+ image_path = str(root / image_path)
|
|
|
+ r = send_request(image_path, image_type)
|
|
|
+ self.assertEqual(item, ResultItem1(status=r['status'],
|
|
|
+ orientation=r['result']['orientation'],
|
|
|
+ type=r['result']['type']['text'],
|
|
|
+ address=r['result']['address']['text']
|
|
|
+ ))
|
|
|
+
|
|
|
+ # 0测试
|
|
|
+
|
|
|
+ # 方向、专业识别错误
|
|
|
+ def test_0_01(self):
|
|
|
+ image_path = '../images/all/0/01.jpeg'
|
|
|
+ self._helper0(image_path, ResultItem0(status='000',
|
|
|
+ orientation=0,
|
|
|
+ name='孙也桐',
|
|
|
+ id='230125199603270066',
|
|
|
+ gender='女',
|
|
|
+ birthplace='黑龙江省宾县',
|
|
|
+ native_place='黑龙江省宾县',
|
|
|
+ blood_type='',
|
|
|
+ religion=''))
|
|
|
+
|
|
|
+ def test_0_02(self):
|
|
|
+ image_path = '../images/all/0/02.jpg'
|
|
|
+ self._helper0(image_path, ResultItem0(status='000',
|
|
|
+ orientation=0,
|
|
|
+ name='宋国军',
|
|
|
+ id='152125610820051',
|
|
|
+ gender='男',
|
|
|
+ birthplace='内蒙古',
|
|
|
+ native_place='河南省陕县',
|
|
|
+ blood_type='A型',
|
|
|
+ religion='无'))
|
|
|
+
|
|
|
+ def test_0_03(self):
|
|
|
+ image_path = '../images/all/0/03.png'
|
|
|
+ self._helper0(image_path, ResultItem0(status='000',
|
|
|
+ orientation=0,
|
|
|
+ name='严倩文',
|
|
|
+ id='360502199810130927',
|
|
|
+ gender='女',
|
|
|
+ birthplace='江西省新余市渝水区',
|
|
|
+ native_place='江西省新余市渝水区',
|
|
|
+ blood_type='不明',
|
|
|
+ religion=''))
|
|
|
+
|
|
|
+ def test_0_04(self):
|
|
|
+ image_path = '../images/all/0/04.jpg'
|
|
|
+ self._helper0(image_path, ResultItem0(status='000',
|
|
|
+ orientation=0,
|
|
|
+ name='李晓令',
|
|
|
+ id='45212219940905067x',
|
|
|
+ gender='男',
|
|
|
+ birthplace='广西横县',
|
|
|
+ native_place='广西横县',
|
|
|
+ blood_type='',
|
|
|
+ religion='无宗教信仰'))
|
|
|
+
|
|
|
+ def test_0_05_0(self):
|
|
|
+ image_path = '../images/all/0/05_0.jpg'
|
|
|
+ self._helper0(image_path, ResultItem0(status='000',
|
|
|
+ orientation=0,
|
|
|
+ name='马宁',
|
|
|
+ id='64038219981016341x',
|
|
|
+ gender='男',
|
|
|
+ birthplace='宁夏灵武市无吴四队',
|
|
|
+ native_place='宁夏',
|
|
|
+ blood_type='B型',
|
|
|
+ religion=''))
|
|
|
+
|
|
|
+ def test_0_05_90(self):
|
|
|
+ image_path = '../images/all/0/05_90.jpg'
|
|
|
+ self._helper0(image_path, ResultItem0(status='000',
|
|
|
+ orientation=1,
|
|
|
+ name='马宁',
|
|
|
+ id='64038219981016341x',
|
|
|
+ gender='男',
|
|
|
+ birthplace='宁夏灵武市无吴四队',
|
|
|
+ native_place='宁夏',
|
|
|
+ blood_type='B型',
|
|
|
+ religion=''))
|
|
|
+
|
|
|
+ def test_0_05_180(self):
|
|
|
+ image_path = '../images/all/0/05_180.jpg'
|
|
|
+ self._helper0(image_path, ResultItem0(status='000',
|
|
|
+ orientation=2,
|
|
|
+ name='马宁',
|
|
|
+ id='64038219981016341x',
|
|
|
+ gender='男',
|
|
|
+ birthplace='宁夏灵武市无吴四队',
|
|
|
+ native_place='宁夏',
|
|
|
+ blood_type='B型',
|
|
|
+ religion=''))
|
|
|
+
|
|
|
+ def test_0_05_270(self):
|
|
|
+ image_path = '../images/all/0/05_270.jpg'
|
|
|
+ self._helper0(image_path, ResultItem0(status='000',
|
|
|
+ orientation=3,
|
|
|
+ name='马宁',
|
|
|
+ id='64038219981016341x',
|
|
|
+ gender='男',
|
|
|
+ birthplace='宁夏灵武市无吴四队',
|
|
|
+ native_place='宁夏',
|
|
|
+ blood_type='B型',
|
|
|
+ religion=''))
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+ def test_1_01_0(self):
|
|
|
+ image_path = '../images/all/1/01_0.png'
|
|
|
+ self._helper1(image_path, ResultItem1(status='000',
|
|
|
+ orientation=0,
|
|
|
+ type='居民家庭户口',
|
|
|
+ address='江西省新余市渝水区胜利南路183号1栋1单元1402室'))
|
|
|
+
|
|
|
+ def test_1_01_90(self):
|
|
|
+ image_path = '../images/all/1/01_90.png'
|
|
|
+ self._helper1(image_path, ResultItem1(status='000',
|
|
|
+ orientation=1,
|
|
|
+ type='居民家庭户口',
|
|
|
+ address='江西省新余市渝水区胜利南路183号1栋1单元1402室'))
|
|
|
+
|
|
|
+ def test_1_01_180(self):
|
|
|
+ image_path = '../images/all/1/01_180.png'
|
|
|
+ self._helper1(image_path, ResultItem1(status='000',
|
|
|
+ orientation=2,
|
|
|
+ type='居民家庭户口',
|
|
|
+ address='江西省新余市渝水区胜利南路183号1栋1单元1402室'))
|
|
|
+
|
|
|
+ def test_1_01_270(self):
|
|
|
+ image_path = '../images/all/1/01_270.png'
|
|
|
+ self._helper1(image_path, ResultItem1(status='000',
|
|
|
+ orientation=3,
|
|
|
+ type='居民家庭户口',
|
|
|
+ address='江西省新余市渝水区胜利南路183号1栋1单元1402室'))
|
|
|
+
|
|
|
+ def test_1_02(self):
|
|
|
+ image_path = '../images/all/1/02.jpg'
|
|
|
+ self._helper1(image_path, ResultItem1(status='000',
|
|
|
+ orientation=0,
|
|
|
+ type='非农业家庭户口',
|
|
|
+ address='东岭东委9组东岭南街解困小区9栋1门201号'))
|
|
|
+
|
|
|
+ def test_1_03(self):
|
|
|
+ image_path = '../images/all/1/03.jpg'
|
|
|
+ self._helper1(image_path, ResultItem1(status='000',
|
|
|
+ orientation=0,
|
|
|
+ type='居民户口',
|
|
|
+ address='黑龙江省桦川县东河乡兴安村1组'))
|
|
|
+
|
|
|
+ def test_1_04(self):
|
|
|
+ image_path = '../images/all/1/04.jpg'
|
|
|
+ self._helper1(image_path, ResultItem1(status='000',
|
|
|
+ orientation=0,
|
|
|
+ type='乡村家庭户',
|
|
|
+ address='宁夏吴忠市利通区金积镇丁家湾子村18040号'))
|
|
|
+
|
|
|
+ def test_1_05(self):
|
|
|
+ image_path = '../images/all/1/05.jpg'
|
|
|
+ self._helper1(image_path, ResultItem1(status='000',
|
|
|
+ orientation=0,
|
|
|
+ type='乡村家庭户',
|
|
|
+ address='宁夏吴忠市利通区郭家桥乡吴桥村四队88号'))
|
|
|
+
|
|
|
+ def test_1_mohu(self):
|
|
|
+ image_path = '../images/all/1/mohu.jpg'
|
|
|
+ self._helper1(image_path, ResultItem1(status='000',
|
|
|
+ orientation=0,
|
|
|
+ type='家庭户',
|
|
|
+ address='甘肃省林洮县康家集乡大头山村羊角湾社'))
|
|
|
+
|
|
|
+ def test_1_special(self):
|
|
|
+ image_path = '../images/all/1/special.jpg'
|
|
|
+ self._helper1(image_path, ResultItem1(status='000',
|
|
|
+ orientation=0,
|
|
|
+ type='居民家庭户',
|
|
|
+ address='安徽省马鞍山市花山区湖北东路479号4栋405室'))
|
|
|
+
|
|
|
+if __name__ == '__main__':
|
|
|
+ unittest.main()
|
zeke-chin
