hr-ocr-idcard/testing/true_test.py

import base64
import unittest
from pathlib import Path

import requests

url = 'http://192.168.199.208:18081'


def send_request(image_path, image_type):
    with open(image_path, 'rb') as f:
        img_str: str = base64.encodebytes(f.read()).decode('utf-8')
        r = requests.post(f'{url}/ocr_system/idcard', json={'image': img_str, 'image_type': image_type})
        print(r.json())
        return r.json()


class TestIdCardAddress(unittest.TestCase):
    def _helper(self, image_path, sta, orient, name, id, ethnicity, gender, birthday, address):
        root = Path(__file__).parent
        image_path = str(root / image_path)
        r = send_request(image_path, '0')
        self.assertEqual(sta, r['status'], f'{image_path} status case error')
        self.assertEqual(orient, r['result']['orientation'], f'{image_path} orientation case error')
        self.assertEqual(name, r['result']['name']['text'], f'{image_path} name case error')
        self.assertEqual(id, r['result']['id']['text'], f'{image_path} id case error')
        self.assertEqual(ethnicity, r['result']['ethnicity']['text'], f'{image_path} ethnicity case error')
        self.assertEqual(gender, r['result']['gender']['text'], f'{image_path} gender case error')
        self.assertEqual(birthday, r['result']['birthday']['text'], f'{image_path} birthday case error')
        self.assertEqual(address, r['result']['address']['text'], f'{image_path} address case error')

    def test_01(self):
        image_path = '../images/ture/01.jpg'
        self._helper(image_path, '000', 0, '宋宝磊', '150430199905051616', '汉', '男', '1999年05月05日',
                     '内蒙古赤峰市敖汉旗四家子镇林家地村唐坊沟七组')

    def test_02(self):
        image_path = '../images/ture/02.jpg'
        self._helper(image_path, '000', 2, '方势文', '360428199610220096', '汉', '男', '1996年10月22日', '江西省九江市都昌县都昌镇沿湖路238号')

    def test_03(self):
        image_path = '../images/ture/03.jpg'
        self._helper(image_path, '000', 0, '彭贤端', '450922199412083669', '汉', '女', '1994年12月08日', '广西陆川县清湖镇塘寨村新屋队62号')

    def test_04(self):
        image_path = '../images/ture/04.png'
        self._helper(image_path, '000', 0, '王晓凤', '150921199910021527', '汉', '女', '1999年10月02日',
                     '呼和浩特市新城区赛马场北路城市维也纳13号楼2单元1303号')

    def test_05(self):
        image_path = '../images/ture/05.jpg'
        self._helper(image_path, '000', 0, '任学东', '152630198501117517', '汉', '男', '1985年01月11日',
                     '内蒙古乌兰察布市察哈尔右翼前旗巴音塔拉镇谢家村22户')

    def test_06(self):
        image_path = '../images/ture/06.jpg'
        self._helper(image_path, '000', 0, '田浩', '640221199702060618', '汉', '男', '1997年02月06日', '宁夏平罗县黄渠桥镇侯家梁村二队29')

    def test_07_0(self):
        image_path = '../images/ture/07_0.jpg'
        self._helper(image_path, '000', 0, '左翔宇', '220204199910123017', '蒙古', '男', '1999年10月12日',
                     '吉林省吉林市船营区鑫安小区2-6-60号')

    def test_07_90(self):
        image_path = '../images/ture/07_90.jpg'
        self._helper(image_path, '000', 1, '左翔宇', '220204199910123017', '蒙古', '男', '1999年10月12日',
                     '吉林省吉林市船营区鑫安小区2-6-60号')

    def test_07_180(self):
        image_path = '../images/ture/07_180.jpg'
        self._helper(image_path, '000', 2, '左翔宇', '220204199910123017', '蒙古', '男', '1999年10月12日',
                     '吉林省吉林市船营区鑫安小区2-6-60号')

    def test_08_0(self):
        image_path = '../images/ture/08_0.jpg'
        self._helper(image_path, '000', 0, '张开天', '622301199710247376', '汉', '男', '1997年10月24日', '甘肃省武威市凉州区九墩乡平乐村四组23号')

    def test_08_180(self):
        image_path = '../images/ture/08_180.jpg'
        self._helper(image_path, '000', 2, '张开天', '622301199710247376', '汉', '男', '1997年10月24日', '甘肃省武威市凉州区九墩乡平乐村四组23号')

    def test_09(self):
        image_path = '../images/ture/09.jpeg'
        self._helper(image_path, '000', 0, '韩毅', '211102199912162010', '汉', '男', '1999年12月16日',
                     '辽宁省盘锦市双台子区胜利街道团结社区15区40号')

    def test_10(self):
        image_path = '../images/ture/10.jpg'
        self._helper(image_path, '000', 0, '曾令权', '432524199306221414', '汉', '男', '1993年06月22日', '湖南省新化县维山乡官庄村陈家冲组21号')



if __name__ == '__main__':
    unittest.main()