| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950 |
- from pathlib import Path
- import unittest
- import config
- from config import send_request
- image_path = 'image'
- # DX_test, DX_sb, DX_test, DX_sb
- envl = 'DX_test'
- url = config.URL[envl]
- token = config.TOKEN[envl]
- class TestOcr(unittest.TestCase):
- def _helper(self, image_path, suffix):
- root = Path(__file__).parent
- image_path = str(root / image_path)
- r = send_request(image_path, suffix, url, token)
- print(r)
- self.assertIn('result', r, f'{suffix} test error')
- def test_bank(self):
- image_path = 'image/bankcard.jpg'
- suffix = '/yhksb/bankcard' if 'sb' in envl else '/yhksbtest/bankcard'
- self._helper(image_path, suffix)
- def test_idcard(self):
- image_path = 'image/idcard.jpg'
- suffix = '/sfzsb/idcard' if 'sb' in envl else '/sfzsbtest/idcard'
- self._helper(image_path, suffix)
- def test_regbook(self):
- image_path = 'image/regbook.jpg'
- suffix = '/hkbsb/regbook' if 'sb' in envl else '/hkbsbtest/regbook'
- self._helper(image_path, suffix)
- def test_schoolcert(self):
- image_path = 'image/schoolcert.jpg'
- suffix = '/xxw/schoolcert' if 'sb' in envl else '/xxwtest/schoolcert'
- self._helper(image_path, suffix)
- def test_cet(self):
- image_path = 'image/cet.jpg'
- suffix = '/cet/cet' if 'sb' in envl else '/cettest/cet'
- self._helper(image_path, suffix)
- def test_blef(self):
- image_path = 'image/blfe.jpg'
- suffix = '/blfe/blfe' if 'sb' in envl else '/blfetest/blfe'
- self._helper(image_path, suffix)
|
