b583b0472f
Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01BJKMLHdxMwSUeYcdkwqX4W
44 lines
1.9 KiB
Python
44 lines
1.9 KiB
Python
"""Адаптер книги -> `FormCandidate` на реальных выгрузках (план 4, задача 5)."""
|
||
|
||
from pathlib import Path
|
||
|
||
import pytest
|
||
|
||
from arm_statements import DocumentType, default_rules
|
||
from src.parser.deadline import Deadline
|
||
from src.parser.detect import is_osv_sheet
|
||
from src.parser.readers import read_book
|
||
from src.parser.rules import load_rules
|
||
from src.parser.statements.adapter import build_candidates, plan_sheets
|
||
|
||
pytestmark = pytest.mark.fixtures
|
||
STATEMENTS = "statements-xlsx"
|
||
NPZ = "npz/2025-Q3/Бухгалтерская отчетность за Январь - Сентябрь 2025 г. (НОВОРОССИЙСКИЙ ПРОКАТНЫЙ ЗАВОД).xlsx"
|
||
UNAGRANDE_BALANCE = "unagrande-group/2025-Q3/ББ. Лист 1 3кв.25.xls"
|
||
|
||
|
||
def _candidates(path: Path): # noqa: ANN202
|
||
sheets, _ = read_book(path.read_bytes(), path.suffix.lstrip(".").upper()) # type: ignore[arg-type]
|
||
osv_rules = load_rules()
|
||
rules = default_rules()
|
||
deadline = Deadline(600.0)
|
||
plan = plan_sheets(sheets, lambda sheet: is_osv_sheet(sheet, osv_rules), rules, deadline)
|
||
return build_candidates(plan.statements, rules, deadline, xlsb=False)
|
||
|
||
|
||
def test_npz_candidates(fixtures_dir: Path) -> None:
|
||
candidates = _candidates(fixtures_dir / STATEMENTS / NPZ)
|
||
assert [(c.type_hint, c.locator.count(",") + 1, len(c.segment_breaks)) for c in candidates] == [
|
||
(DocumentType.BALANCE, 5, 4),
|
||
(DocumentType.INCOME, 3, 2),
|
||
]
|
||
codes = {w.code for c in candidates for w in c.warnings}
|
||
assert not codes & {"SHEET_NOT_JOINED", "CONTINUATION_TYPE_CONFLICT"}
|
||
|
||
|
||
def test_unagrande_title_block_header(fixtures_dir: Path) -> None:
|
||
candidates = _candidates(fixtures_dir / STATEMENTS / UNAGRANDE_BALANCE)
|
||
header = candidates[0].header
|
||
assert header.okopf is not None and header.okopf.text == "12267"
|
||
assert header.inn is not None and "7728757141" in header.inn.text
|