import unittest import base64 from dataclasses import dataclass from pathlib import Path import requests # url = 'http://localhost:8080' url = 'http://aihub.digitalyili.com/aiSquare/openApi/reasoning-services' header = { 'Authorization': 'Bearer dcae8cc6-0e49-4db8-a2d2-94ef84da3636' } # 带线框识别比较差 例:6.27_02.png def send_request(image_path, image_type='1'): with open(image_path, 'rb') as f: img_str: str = base64.encodebytes(f.read()).decode('utf-8') r = requests.post(f'{url}/rlocrxm/xxw/schoolcert', json={'image': img_str, 'image_type': image_type}, headers=header) print(r.json()) return r.json() @dataclass class ResultItem: status: str orientation: int name: str gender: str admission_time: str education_time: str education_level: str education_type: str learning_type: str school: str major: str number: str class TestSchoolCertOcr(unittest.TestCase): def _helper(self, image_path, item: ResultItem, image_type='1'): root = Path(__file__).parent image_path = str(root / image_path) r = send_request(image_path, image_type) self.assertEqual(item, ResultItem(status=r['status'], orientation=r['result']['orientation'], name=r['result']['name']['text'], gender=r['result']['gender']['text'], admission_time=r['result']['admission_time']['text'], education_time=r['result']['education_time']['text'], education_type=r['result']['education_type']['text'], education_level=r['result']['education_level']['text'], learning_type=r['result']['learning_type']['text'], school=r['result']['school']['text'], major=r['result']['major']['text'], number=r['result']['number']['text'] )) # 0测试 # 方向、专业识别错误 def test_0_01_0(self): image_path = '../images/all/0/01_0.jpg' self._helper(image_path, ResultItem(status='000', orientation=0, name='王海龙', gender='男', admission_time='2018年09月05日', education_time='2021年06月30日', education_level='专科', education_type='普通高等教育', learning_type='普通全日制', school='新疆农业职业技术学院', major='食品营养与检测', number='')) # 多种错误 def test_0_01_90(self): image_path = '../images/all/0/01_90.jpg' self._helper(image_path, ResultItem(status='000', orientation=1, name='王海龙', gender='男', admission_time='2018年09月05日', education_time='2021年06月30日', education_level='专科', education_type='普通高等教育', learning_type='普通全日制', school='新疆农业职业技术学院', major='食品营养与检测', number='')) # 方向、毕业时间识别错误 def test_0_01_180(self): image_path = '../images/all/0/01_180.jpg' self._helper(image_path, ResultItem(status='000', orientation=2, name='王海龙', gender='男', admission_time='2018年09月05日', education_time='2021年06月30日', education_level='专科', education_type='普通高等教育', learning_type='普通全日制', school='新疆农业职业技术学院', major='食品营养与检测', number='')) # 方向、专业识别错误 def test_0_01_270(self): image_path = '../images/all/0/01_270.jpg' self._helper(image_path, ResultItem(status='000', orientation=3, name='王海龙', gender='男', admission_time='2018年09月05日', education_time='2021年06月30日', education_level='专科', education_type='普通高等教育', learning_type='普通全日制', school='新疆农业职业技术学院', major='食品营养与检测', number='')) # 方向识别错误 def test_0_02(self): image_path = '../images/all/0/02.jpg' self._helper(image_path, ResultItem(status='000', orientation=0, name='张博渊', gender='男', admission_time='2018年09月12日', education_time='2022年07月01日', education_level='本科', education_type='普通高等教育', learning_type='普通全日制', school='内蒙古科技大学', major='应用化学', number='')) # 姓名、学校识别错误 def test_0_03(self): image_path = '../images/all/0/03.jpg' self._helper(image_path, ResultItem(status='000', orientation=0, name='徐志林', gender='男', admission_time='2019年09月01日', education_time='2021年07月01日', education_level='专科', education_type='普通', learning_type='普通全日制', school='四川工商职业技术学院', major='食品生物技术', number='')) # 入学时间识别错误 def test_0_05_0(self): image_path = '../images/all/0/05_0.jpg' self._helper(image_path, ResultItem(status='000', orientation=0, name='金泽丞', gender='男', admission_time='2020年09月01日', education_time='2022年07月10日', education_level='本科', education_type='普通', learning_type='普通全日制', school='内蒙古科技大学', major='自动化', number='')) # 入学时间识别错误 def test_0_05_90(self): image_path = '../images/all/0/05_90.jpg' self._helper(image_path, ResultItem(status='000', orientation=1, name='金泽丞', gender='男', admission_time='2020年09月01日', education_time='2022年07月10日', education_level='本科', education_type='普通', learning_type='普通全日制', school='内蒙古科技大学', major='自动化', number='')) # 多种错误 def test_0_05_180(self): image_path = '../images/all/0/05_180.jpg' self._helper(image_path, ResultItem(status='000', orientation=2, name='金泽丞', gender='男', admission_time='2020年09月01日', education_time='2022年07月10日', education_level='本科', education_type='普通', learning_type='普通全日制', school='内蒙古科技大学', major='自动化', number='')) # 入学时间识别错误 def test_0_05_270(self): image_path = '../images/all/0/05_270.jpg' self._helper(image_path, ResultItem(status='000', orientation=3, name='金泽丞', gender='男', admission_time='2020年09月01日', education_time='2022年07月10日', education_level='本科', education_type='普通', learning_type='普通全日制', school='内蒙古科技大学', major='自动化', number='')) # 学校识别错误 def test_0_06(self): image_path = '../images/all/0/06.jpg' self._helper(image_path, ResultItem(status='000', orientation=0, name='张豪', gender='男', admission_time='2020年10月10日', education_time='2022年06月30日', education_level='专科', education_type='普通', learning_type='普通全日制', school='四川工商职业技术学院', major='食品生物技术', number='')) # 1测试 # 方向、姓名错误 def test_1_01(self): image_path = '../images/all/1/01.jpg' self._helper(image_path, ResultItem(status='000', orientation=0, name='陈默涵', gender='女', admission_time='2018年09月05日', education_time='2021年06月25日', education_level='专科', education_type='普通高等教育', learning_type='普通全日制', school='新疆农业职业技术学院', major='食品营养与检测', number='109951202106002702'), '0') # 多种错误 def test_1_02_0(self): image_path = '../images/all/1/02_0.png' self._helper(image_path, ResultItem(status='000', orientation=0, name='张敏', gender='男', admission_time='2016年09月10日', education_time='2020年06月30日', education_level='本科', education_type='普通高等教育', learning_type='普通全日制', school='大连科技学院', major='机械设计制造及其自动化', number='132071202005100273'), '0') # 多种错误 def test_1_02_90(self): image_path = '../images/all/1/02_90.png' self._helper(image_path, ResultItem(status='000', orientation=1, name='张敏', gender='男', admission_time='2016年09月10日', education_time='2020年06月30日', education_level='本科', education_type='普通高等教育', learning_type='普通全日制', school='大连科技学院', major='机械设计制造及其自动化', number='132071202005100273'), '0') # 多种错误 def test_1_02_180(self): image_path = '../images/all/1/02_180.png' self._helper(image_path, ResultItem(status='000', orientation=2, name='张敏', gender='男', admission_time='2016年09月10日', education_time='2020年06月30日', education_level='本科', education_type='普通高等教育', learning_type='普通全日制', school='大连科技学院', major='机械设计制造及其自动化', number='132071202005100273'), '0') # 多种错误 def test_1_02_270(self): image_path = '../images/all/1/02_270.png' self._helper(image_path, ResultItem(status='000', orientation=3, name='张敏', gender='男', admission_time='2016年09月10日', education_time='2020年06月30日', education_level='本科', education_type='普通高等教育', learning_type='普通全日制', school='大连科技学院', major='机械设计制造及其自动化', number='132071202005100273'), '0') def test_1_03(self): image_path = '../images/all/1/03.JPG' self._helper(image_path, ResultItem(status='000', orientation=0, name='薛峰', gender='男', admission_time='2009年09月01日', education_time='2013年06月30日', education_level='本科', education_type='普通高等教育', learning_type='普通全日制', school='武汉纺织大学外经贸学院', major='高分子材料与工程', number='132401201305799592'), '0') # 多种错误 def test_1_04(self): image_path = '../images/all/1/04.jpg' self._helper(image_path, ResultItem(status='000', orientation=0, name='许星皓', gender='男', admission_time='', education_time='2012年12月30日', education_level='专科', education_type='高等教育自学考试', learning_type='', school='海南科技职业学院', major='数控技术应用（专科）', number='66460183101004424'), '0') # 方向错误 def test_1_05(self): image_path = '../images/all/1/05.jpg' self._helper(image_path, ResultItem(status='000', orientation=0, name='李婷婷', gender='女', admission_time='2015年09月01日', education_time='2019年07月01日', education_level='本科', education_type='普通高等教育', learning_type='普通全日制', school='广西财经学院', major='商务英语', number='115481201905007155'), '0') # 2测试 # miss orientation、number def test_2_01_0(self): image_path = '../images/all/2/01_0.png' self._helper(image_path, ResultItem(status='000', orientation=0, name='肖亮', gender='男', admission_time='2012年9月8日', education_time='2018年6月28日', education_level='本科', education_type='普通高等教育', learning_type='普通全日制', school='济南大学泉城学院', major='市场营销', number='140021201805000882'), '2') # 多种错误 def test_2_01_90(self): image_path = '../images/all/2/01_90.png' self._helper(image_path, ResultItem(status='000', orientation=1, name='肖亮', gender='男', admission_time='2012年9月8日', education_time='2018年6月28日', education_level='本科', education_type='普通高等教育', learning_type='普通全日制', school='济南大学泉城学院', major='市场营销', number='140021201805000882'), '2') # 多种错误 def test_2_01_180(self): image_path = '../images/all/2/01_180.png' self._helper(image_path, ResultItem(status='000', orientation=2, name='肖亮', gender='男', admission_time='2012年9月8日', education_time='2018年6月28日', education_level='本科', education_type='普通高等教育', learning_type='普通全日制', school='济南大学泉城学院', major='市场营销', number='140021201805000882'), '2') # 多种错误 def test_2_01_270(self): image_path = '../images/all/2/01_180.png' self._helper(image_path, ResultItem(status='000', orientation=3, name='肖亮', gender='男', admission_time='2012年9月8日', education_time='2018年6月28日', education_level='本科', education_type='普通高等教育', learning_type='普通全日制', school='济南大学泉城学院', major='市场营销', number='140021201805000882'), '2') # miss learning_type、number def test_2_02(self): image_path = '../images/all/2/02.jpeg' self._helper(image_path, ResultItem(status='000', orientation=0, name='提姆·库克', gender='男', admission_time='2001年09月01日', education_time='2004年07月01日', education_level='本科', education_type='高等教育', learning_type='', school='河北工业大学', major='人工智能之机器学习', number='100801200405001235'), '2') # 名字错误 def test_2_03(self): image_path = '../images/all/2/03.png' self._helper(image_path, ResultItem(status='000', orientation=0, name='胡磊', gender='男', admission_time='2012年9月1日', education_time='2016年7月1日', education_level='本科', education_type='普通高等教育', learning_type='普通全日制', school='红河学院', major='经济学', number='106871201605000620'), '2') # 正确 def test_2_04(self): image_path = '../images/all/2/04.png' self._helper(image_path, ResultItem(status='000', orientation=0, name='孟煜翔', gender='男', admission_time='2016年9月3日', education_time='2019年6月24日', education_level='专科', education_type='普通高等教育', learning_type='普通全日制', school='山东农业工程学院', major='环境艺术设计', number='144391201906001607'), '2') # 方向错误 def test_2_05(self): image_path = '../images/all/2/05.jpg' self._helper(image_path, ResultItem(status='000', orientation=0, name='黄浦村', gender='男', admission_time='2011年9月1日', education_time='2014年7月1日', education_level='专科', education_type='普通高等教育', learning_type='普通全日制', school='江西工程学院', major='装潢艺术设计', number='127661201406001009'), '2')