-
Notifications
You must be signed in to change notification settings - Fork 270
Expand file tree
/
Copy pathtest_ocr.py
More file actions
34 lines (25 loc) · 1014 Bytes
/
test_ocr.py
File metadata and controls
34 lines (25 loc) · 1014 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
# -*- coding: UTF-8 -*-
"""OCR功能测试模块。
该模块包含对python-office库中OCR相关功能的单元测试。
@学习网站 ：https://www.python-office.com
@读者群 ：https://www.python4office.cn/wechat-group/
@作者 ：B站/抖音/微博/小红书/公众号，都叫：程序员晚枫，微信：wfdev7
@代码日期 ：2025/7/2 22:45
@本段代码的视频说明 ：
"""
import os
import unittest
from loguru import logger
from office.api.ocr import VatInvoiceOCR2Excel
class TestPDF(unittest.TestCase):
"""OCR功能测试类。
该类包含对OCR相关API的单元测试方法。
"""
def setUp(self):
self.id = os.getenv("SecretId", None)
self.key = os.getenv("SecretKey", None)
def test_VatInvoiceOCR2Excel(self):
logger.info(self.id)
logger.info(self.key)
VatInvoiceOCR2Excel(input_path=r'..\test_files\ocr\img.png', output_path=r'./output/ocr',
id=self.id, key=self.key)