fix: 补充 __init__.py 导出符号,修复测试导入错误
- src/core/extraction/__init__.py: 导出 _image_to_base64 - src/core/matching/__init__.py: 导出所有内部函数(_build_invoice_summary, _build_payment_records, _invoices_to_records, _match, _match_by_filename, _match_one_to_many, _match_one_to_one, _relative_tolerance, _safe_float) - src/infra/documents/__init__.py: 导出 INVOICE_LEVEL_COLUMNS, PAYMENT_RECORD_COLUMNS - tests/test_extractor.py: 修正 monkeypatch 路径为模块级引用
This commit is contained in:
@@ -11,6 +11,7 @@ from .extractor import (
|
||||
)
|
||||
from .llm_extractor import (
|
||||
CACHE_DIR_NAME,
|
||||
_image_to_base64,
|
||||
build_extraction_user_message,
|
||||
extract_document,
|
||||
extract_normal_info,
|
||||
@@ -25,6 +26,7 @@ from .llm_extractor import (
|
||||
|
||||
__all__ = [
|
||||
"CACHE_DIR_NAME",
|
||||
"_image_to_base64",
|
||||
"build_extraction_user_message",
|
||||
"extract_document",
|
||||
"extract_normal_info",
|
||||
|
||||
@@ -3,6 +3,28 @@
|
||||
提供发票与支付记录的金额匹配功能。
|
||||
"""
|
||||
|
||||
from .matcher import match_invoices_to_cards
|
||||
from .matcher import (
|
||||
_build_invoice_summary,
|
||||
_build_payment_records,
|
||||
_invoices_to_records,
|
||||
_match,
|
||||
_match_by_filename,
|
||||
_match_one_to_many,
|
||||
_match_one_to_one,
|
||||
_relative_tolerance,
|
||||
_safe_float,
|
||||
match_invoices_to_cards,
|
||||
)
|
||||
|
||||
__all__ = ["match_invoices_to_cards"]
|
||||
__all__ = [
|
||||
"_build_invoice_summary",
|
||||
"_build_payment_records",
|
||||
"_invoices_to_records",
|
||||
"_match",
|
||||
"_match_by_filename",
|
||||
"_match_one_to_many",
|
||||
"_match_one_to_one",
|
||||
"_relative_tolerance",
|
||||
"_safe_float",
|
||||
"match_invoices_to_cards",
|
||||
]
|
||||
|
||||
@@ -10,6 +10,8 @@ from .consumable import (
|
||||
)
|
||||
from .invoice import (
|
||||
CACHE_DIR_NAME,
|
||||
INVOICE_LEVEL_COLUMNS,
|
||||
PAYMENT_RECORD_COLUMNS,
|
||||
classify_invoice_batch,
|
||||
load_csv,
|
||||
load_invoice_csv,
|
||||
@@ -20,6 +22,8 @@ from .invoice import (
|
||||
|
||||
__all__ = [
|
||||
"CACHE_DIR_NAME",
|
||||
"INVOICE_LEVEL_COLUMNS",
|
||||
"PAYMENT_RECORD_COLUMNS",
|
||||
"classify_invoice_batch",
|
||||
"load_csv",
|
||||
"load_invoice_csv",
|
||||
|
||||
@@ -129,7 +129,7 @@ class TestExtractInvoices:
|
||||
}
|
||||
]
|
||||
|
||||
monkeypatch.setattr("src.core.matching.matcher.match_invoices_to_cards", fake_match)
|
||||
monkeypatch.setattr("src.core.extraction.extractor.match_invoices_to_cards", fake_match)
|
||||
|
||||
records, apps, groups = extract_invoices(str(tmp_path))
|
||||
|
||||
@@ -194,7 +194,7 @@ class TestExtractInvoices:
|
||||
},
|
||||
]
|
||||
|
||||
monkeypatch.setattr("src.core.matching.matcher.match_invoices_to_cards", fake_match)
|
||||
monkeypatch.setattr("src.core.extraction.extractor.match_invoices_to_cards", fake_match)
|
||||
|
||||
records, apps, groups = extract_invoices(str(tmp_path))
|
||||
|
||||
@@ -241,7 +241,7 @@ class TestExtractInvoices:
|
||||
}
|
||||
]
|
||||
|
||||
monkeypatch.setattr("src.core.matching.matcher.match_invoices_to_cards", fake_match)
|
||||
monkeypatch.setattr("src.core.extraction.extractor.match_invoices_to_cards", fake_match)
|
||||
|
||||
records, apps, groups = extract_invoices(str(tmp_path))
|
||||
|
||||
@@ -283,7 +283,7 @@ class TestExtractInvoices:
|
||||
}
|
||||
]
|
||||
|
||||
monkeypatch.setattr("src.core.matching.matcher.match_invoices_to_cards", fake_match)
|
||||
monkeypatch.setattr("src.core.extraction.extractor.match_invoices_to_cards", fake_match)
|
||||
|
||||
records, apps, groups = extract_invoices(str(tmp_path))
|
||||
|
||||
@@ -324,7 +324,7 @@ class TestExtractInvoices:
|
||||
}
|
||||
]
|
||||
|
||||
monkeypatch.setattr("src.core.matching.matcher.match_invoices_to_cards", fake_match)
|
||||
monkeypatch.setattr("src.core.extraction.extractor.match_invoices_to_cards", fake_match)
|
||||
|
||||
records, apps, groups = extract_invoices(str(tmp_path))
|
||||
|
||||
|
||||
Reference in New Issue
Block a user