hr-ocr-idcard/testing/ocr_test.py

import unittest
import base64
import requests

url = 'http://localhost:8080'


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 TestIdCardOcr(unittest.TestCase):
    def test_it_works(self):
        self.assertEqual(1, 1, 'it not works')

    def test_test(self):
        image_path = './images/test.jpeg'
        r = send_request(image_path, '0')
        self.assertEqual(r['status'], '000', 'status case error')
        self.assertEqual(r['result']['orientation'], 0, 'orientation case error')
        self.assertEqual(r['result']['name']['text'], '阎妍', 'name case error')
        self.assertEqual(r['result']['id']['text'], '410802198701190046', 'id case error')

    def test_front_90(self):
        image_path = './images/front-90.jpg'
        r = send_request(image_path, '0')
        self.assertEqual(r['status'], '000', 'status case error')
        self.assertEqual(r['result']['orientation'], 1, 'orientation case error')
        self.assertEqual(r['result']['name']['text'], '刘新建', 'name case error')
        self.assertEqual(r['result']['id']['text'], '412902197906291716', 'id case error')
        self.assertEqual(r['result']['birthday']['text'], '1979年6月29日', 'birthday case error')
        self.assertEqual(r['result']['address_province']['text'], '河南省', 'address_province case error')
        self.assertEqual(r['result']['address_city']['text'], '邓州市', 'address_city case error')
        self.assertEqual(r['result']['address_region']['text'], '白牛乡', 'address_region case error')
        self.assertEqual(r['result']['address_detail']['text'], '故事村故事桥111号', 'address_detail case error')

    def test_front_180(self):
        image_path = './images/front-180.png'
        r = send_request(image_path, '0')
        self.assertEqual(r['status'], '000', 'status case error')
        self.assertEqual(r['result']['orientation'], 2, 'orientation case error')
        self.assertEqual(r['result']['name']['text'], '阎妍', 'name case error')
        self.assertEqual(r['result']['id']['text'], '410802198701190046', 'id case error')

    def test_front_270(self):
        image_path = './images/front-270.png'
        r = send_request(image_path, '0')
        self.assertEqual(r['status'], '000', 'status case error')
        self.assertEqual(r['result']['orientation'], 3, 'orientation case error')
        self.assertEqual(r['result']['name']['text'], '阎妍', 'name case error')
        self.assertEqual(r['result']['id']['text'], '410802198701190046', 'id case error')

    def test_long_exp(self):
        image_path = './images/long-exp.png'
        r = send_request(image_path, '1')
        self.assertEqual(r['status'], '000', 'status case error')
        self.assertEqual(r['result']['expire_date']['text'], '20180227-长期', 'expire date case error')

    def test_tilt1(self):
        image_path = './images/tilt.jpg'
        r = send_request(image_path, '0')
        self.assertEqual(r['status'], '000', 'status case error')
        self.assertEqual(r['result']['orientation'], 0 or 1, 'orientation case error')

    def test_tilt2(self):
        image_path = './images/tilt.jpg'
        r = send_request(image_path, '0')
        self.assertEqual(r['result']['name']['text'], '奥巴马', 'name case error')
        self.assertEqual(r['result']['birthday']['text'], '1961年8月4日', 'birthday case error')

    def test_tilt3(self):
        image_path = './images/tilt.jpg'
        r = send_request(image_path, '0')
        self.assertEqual(r['result']['birthday']['text'], '1961年8月4日', 'birthday case error')

    def test_vague(self):
        image_path = './images/vague.jpg'
        r = send_request(image_path, '0')
        self.assertEqual(r['status'], '101', 'status case error')

    def test_bank(self):
        image_path = './images/bank.jpg'
        r = send_request(image_path, '0')
        self.assertEqual(r['status'], '101', 'status case error')

    def test_back(self):
        image_path = './images/back.jpeg'
        r = send_request(image_path, '1')
        self.assertEqual(r['status'], '000', 'status case error')
        self.assertEqual(r['result']['orientation'], 0, 'orientation case error')
        self.assertEqual(r['result']['expire_date']['text'], '20220114-20420114', 'expire_date case error')

    def test_back_270(self):
        image_path = './images/back-270.jpg'
        r = send_request(image_path, '1')
        self.assertEqual(r['status'], '000', 'status case error')
        self.assertEqual(r['result']['orientation'], 3, 'orientation case error')
        self.assertEqual(r['result']['expire_date']['text'], '20220114-20420114', 'expire_date case error')

    def test_back_180(self):
        image_path = './images/back-180.jpg'
        r = send_request(image_path, '1')
        self.assertEqual(r['status'], '000', 'status case error')
        self.assertEqual(r['result']['orientation'], 2, 'orientation case error')
        self.assertEqual(r['result']['expire_date']['text'], '20220114-20420114', 'expire_date case error')

    def test_back_90(self):
        image_path = './images/back-90.jpg'
        r = send_request(image_path, '1')
        self.assertEqual(r['status'], '000', 'status case error')
        self.assertEqual(r['result']['orientation'], 1, 'orientation case error')
        self.assertEqual(r['result']['expire_date']['text'], '20220114-20420114', 'expire_date case error')

    def test_back_vague(self):
        image_path = './images/back-vague.jpeg'
        r = send_request(image_path, '1')
        self.assertEqual(r['status'], '000', 'status case error')
        self.assertEqual(r['result']['orientation'], 0, 'orientation case error')
        self.assertEqual(r['result']['expire_date']['text'], '20120724-20220724', 'expire_date case error')