add test case

testing/ocr_test.py

@@ -17,17 +17,42 @@ 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'
@@ -35,5 +60,67 @@ class TestIdCardOcr(unittest.TestCase):
         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')
+