100 lines
3.1 KiB
Python
100 lines
3.1 KiB
Python
"""Unit-тесты services/classify.py — фильтры + JSON-парсинг + выжимка."""
|
|
import json
|
|
|
|
import pytest
|
|
|
|
from services.classify import (
|
|
_is_garbage_by_filename,
|
|
_is_garbage_by_header,
|
|
_safe_json_parse,
|
|
_smart_extract,
|
|
)
|
|
|
|
|
|
class TestGarbageFilename:
|
|
def test_invoice(self):
|
|
assert _is_garbage_by_filename("счет-фактура №123.docx") is True
|
|
|
|
def test_act(self):
|
|
assert _is_garbage_by_filename("Акт сверки.pdf") is True
|
|
|
|
def test_upd(self):
|
|
assert _is_garbage_by_filename("УПД-2025.docx") is True
|
|
|
|
def test_not_garbage(self):
|
|
assert _is_garbage_by_filename("договор.docx") is False
|
|
assert _is_garbage_by_filename("спецификация.pdf") is False
|
|
|
|
|
|
class TestGarbageHeader:
|
|
def test_invoice_header(self):
|
|
assert _is_garbage_by_header("СЧЕТ-ФАКТУРА № 123 от 01.01.2026") is True
|
|
|
|
def test_act_header(self):
|
|
assert _is_garbage_by_header("АКТ ОКАЗАННЫХ УСЛУГ за май") is True
|
|
|
|
def test_not_garbage(self):
|
|
assert _is_garbage_by_header("ДОГОВОР № 03700_1 об оказании услуг") is False
|
|
|
|
def test_upd_mention_in_contract_is_not_garbage(self):
|
|
text = "Договор № 03700. Оплата производится на основании УПД и счета."
|
|
assert _is_garbage_by_header(text) is False
|
|
|
|
def test_upd_document_header_is_garbage(self):
|
|
assert _is_garbage_by_header("УПД № 123 от 01.01.2026") is True
|
|
|
|
|
|
class TestSafeJsonParse:
|
|
def test_valid(self):
|
|
d, fix = _safe_json_parse('{"a":1}')
|
|
assert d == {"a": 1}
|
|
assert fix is False
|
|
|
|
def test_markdown(self):
|
|
d, fix = _safe_json_parse('```json\n{"a":1}\n```')
|
|
assert d == {"a": 1}
|
|
assert fix is False
|
|
|
|
def test_chatter(self):
|
|
d, _ = _safe_json_parse('Вот ответ: {"a":1}. Спасибо!')
|
|
assert d == {"a": 1}
|
|
|
|
def test_trailing_comma(self):
|
|
d, fix = _safe_json_parse('{"a":1,}')
|
|
assert d == {"a": 1}
|
|
assert fix is True
|
|
|
|
def test_trailing_comma_in_list(self):
|
|
d, fix = _safe_json_parse('{"a":[1,2,]}')
|
|
assert d == {"a": [1, 2]}
|
|
assert fix is True
|
|
|
|
def test_empty_raises(self):
|
|
with pytest.raises(ValueError):
|
|
_safe_json_parse("")
|
|
|
|
def test_none_raises(self):
|
|
with pytest.raises(ValueError):
|
|
_safe_json_parse(None)
|
|
|
|
|
|
class TestSmartExtract:
|
|
def test_paragraphs(self):
|
|
ej = [{"type": "paragraph", "text": "Договор № 03700"}]
|
|
out = _smart_extract(ej)
|
|
assert "Договор" in out
|
|
|
|
def test_table(self):
|
|
ej = [{"type": "table", "rows": [["Аренда", "50000"]]}]
|
|
out = _smart_extract(ej)
|
|
assert "Аренда" in out
|
|
|
|
def test_string_json(self):
|
|
ej = json.dumps([{"type": "paragraph", "text": "Спецификация"}])
|
|
out = _smart_extract(ej)
|
|
assert "Спецификация" in out
|
|
|
|
def test_empty(self):
|
|
assert _smart_extract(None) == ""
|
|
assert _smart_extract("") == ""
|