9dd568ce44
I-1: «(4,200) руб», «1,234(5,678)» и т. п. разбираются как прежде; I-2/I-3: спека и README; M-1..M-3: docstring, комментарий регулярки, тесты крайних случаев. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
70 lines
3.9 KiB
Python
70 lines
3.9 KiB
Python
"""Суммы из OCR-ячеек (§11.1 п. 7): перенос `_parse_amount`/`_is_blank` старого ядра."""
|
||
|
||
from decimal import Decimal
|
||
|
||
import pytest
|
||
|
||
from src.normalizer.ocr.amounts import ParsedAmount, is_blank, parse_amount
|
||
|
||
|
||
@pytest.mark.parametrize(
|
||
("raw", "amount", "verified"),
|
||
[
|
||
("13 070", Decimal("13070"), True),
|
||
("(6032409)", Decimal("-6032409"), True),
|
||
("(7744536}", Decimal("7744536"), False), # закрывающая скобка не распознана: минус не ставится (решение №12)
|
||
("1 234,56", Decimal("1234.56"), True),
|
||
("1.234.567", Decimal("1234567"), False), # несколько точек — разделители тысяч; текст изменён → не verified
|
||
("30127", Decimal("30127"), True),
|
||
("211(1 |", Decimal("2111"), False),
|
||
("не ****", None, False),
|
||
("", None, True),
|
||
("--", None, True), # прочерк: ничего не отброшено, суммы нет (NULL)
|
||
(":", None, False),
|
||
# несколько точек: удаляются ВСЕ точки, десятичная часть теряется (`1.234,56` — не 1234.56), всегда не verified
|
||
("1.234,56", Decimal("123456"), False),
|
||
("1,234.56", Decimal("123456"), False),
|
||
# запятая-разделитель тысяч (AURORA-1405, scan-gateway#35): удаляется до замены запятой на точку, verified=True
|
||
("1,234", Decimal("1234"), True),
|
||
("4,200", Decimal("4200"), True),
|
||
("(4,200)", Decimal("-4200"), True),
|
||
("1,234,567", Decimal("1234567"), True),
|
||
("-4,200", Decimal("-4200"), True),
|
||
("+4,200", Decimal("4200"), True),
|
||
("(-4,200)", Decimal("-4200"), True),
|
||
("4,2", Decimal("4.2"), True),
|
||
("12,34", Decimal("12.34"), True),
|
||
("1,23", Decimal("1.23"), True),
|
||
# не тысячи: больше двух знаков после запятой остаётся десятичным — `Money` отсекает `document._value` (№30)
|
||
("0,125", Decimal("0.125"), True),
|
||
("12,3456", Decimal("12.3456"), True),
|
||
("4,200,", Decimal("4200"), False), # хвостовая запятая: правило не применяется (две точки удалены, как раньше)
|
||
("(4,200}", Decimal("4.200"), False), # нет пары скобок: правило не применяется, минус не ставится
|
||
# скобки не на всю строку — правило тысяч не применяется, прежний разбор (запятая → точка)
|
||
("(4,200) руб", Decimal("-4.200"), False),
|
||
("abc(4,200)", Decimal("-4.200"), False),
|
||
("(4,200)x", Decimal("-4.200"), False),
|
||
("(4,200)(1,000)", Decimal("-4.200"), False),
|
||
("1,234(5,678)", Decimal("-5.678"), False),
|
||
("((4,200))", Decimal("-4.200"), True),
|
||
("01,234", Decimal("1.234"), True), # ведущий ноль — не тысячи
|
||
("4,2000", Decimal("4.2000"), True),
|
||
("1, 234", Decimal("1234"), True), # пробелы убираются первыми
|
||
# №13: скобки без содержимого — не сумма, но и не «неразобранный текст»
|
||
("()", None, True),
|
||
("( )", None, True),
|
||
("(.)", None, True),
|
||
("(+5)", Decimal("-5"), True), # в скобках знак заменяется минусом
|
||
],
|
||
)
|
||
def test_parse_amount(raw: str, amount: Decimal | None, verified: bool) -> None:
|
||
assert parse_amount(raw) == ParsedAmount(amount, verified)
|
||
|
||
|
||
@pytest.mark.parametrize(
|
||
("raw", "expected"),
|
||
[("", True), ("-", True), ("—", True), (" – ", True), ("0", False), ("x", False)],
|
||
)
|
||
def test_is_blank(raw: str, expected: bool) -> None:
|
||
assert is_blank(raw) is expected
|