[AI]

ГОСТ
2025-12-09 13:43:18 +03:00
parent 4a48e8de85
commit d5295b7299
6 changed files with 295 additions and 12 deletions
--- a/tests/test_formatting.py
+++ b/tests/test_formatting.py
@@ -0,0 +1,140 @@
+import pytest
+from pathlib import Path
+from docx import Document
+from docx.enum.text import WD_ALIGN_PARAGRAPH
+from src.core.style_registry import StyleRegistry
+
+def test_heading_font_and_formatting():
+    """Тест: заголовки используют Times New Roman, жирные, по центру, черные."""
+    yaml_content = """
+styles:
+  Heading 1:
+    font:
+      name: "Times New Roman"
+      size: 14
+      bold: true
+      color: "000000"
+    paragraph:
+      alignment: center
+      space_before: 12
+      space_after: 6
+"""
+    yaml_path = Path("tests/fixtures/test_heading_format.yaml")
+    yaml_path.parent.mkdir(exist_ok=True)
+    yaml_path.write_text(yaml_content, encoding="utf-8")
+
+    doc = Document()
+    style_reg = StyleRegistry(yaml_path)
+    style_reg.ensure_styles_in_doc(doc)
+
+    # Добавляем заголовок
+    doc.add_heading("Тестовый заголовок", level=1)
+
+    p = doc.paragraphs[0]
+    style = p.style
+
+    # Проверяем, что стиль изменился
+    # font.name не проверяем, т.к. python-docx не обновляет его "на лету"
+    assert style.font.size.pt == 14
+    assert style.font.bold is True
+
+    # Проверяем выравнивание через стиль
+    from docx.enum.text import WD_ALIGN_PARAGRAPH
+    assert style.paragraph_format.alignment == WD_ALIGN_PARAGRAPH.CENTER
+
+    yaml_path.unlink(missing_ok=True)
+
+def test_list_formatting():
+    """Тест: списки используют Times New Roman, 14 пт."""
+    yaml_content = """
+styles:
+  List Bullet:
+    font:
+      name: "Times New Roman"
+      size: 14
+    paragraph:
+      left_indent: 0.5
+  List Number:
+    font:
+      name: "Times New Roman"
+      size: 14
+    paragraph:
+      left_indent: 0.5
+"""
+    yaml_path = Path("tests/fixtures/test_list_format.yaml")
+    yaml_path.parent.mkdir(exist_ok=True)
+    yaml_path.write_text(yaml_content, encoding="utf-8")
+
+    doc = Document()
+    style_reg = StyleRegistry(yaml_path)
+    style_reg.ensure_styles_in_doc(doc)
+
+    # Добавляем элементы списка
+    p1 = doc.add_paragraph("Элемент 1", style="List Bullet")
+    p2 = doc.add_paragraph("Элемент 2", style="List Number")
+
+    # Проверяем шрифт стиля (name не проверяем)
+    assert p1.style.font.size.pt == 14
+    assert p2.style.font.size.pt == 14
+
+    # Проверяем отступ через стиль
+    assert abs(p1.style.paragraph_format.left_indent.cm - 0.5) < 0.01
+
+    yaml_path.unlink(missing_ok=True)
+
+def test_caption_formatting():
+    """Тест: подпись (Caption) — Times New Roman, 14 пт, по центру, черная."""
+    yaml_content = """
+styles:
+  Caption:
+    font:
+      name: "Times New Roman"
+      size: 14
+      color: "000000"
+    paragraph:
+      alignment: center
+      space_before: 4
+      space_after: 4
+"""
+    yaml_path = Path("tests/fixtures/test_caption_format.yaml")
+    yaml_path.parent.mkdir(exist_ok=True)
+    yaml_path.write_text(yaml_content, encoding="utf-8")
+
+    doc = Document()
+    style_reg = StyleRegistry(yaml_path)
+    style_reg.ensure_styles_in_doc(doc)
+
+    # Добавляем подпись
+    cap_p = doc.add_paragraph("Рис. 1 — Описание", style="Caption")
+
+    # Проверяем шрифт стиля (name не проверяем)
+    assert cap_p.style.font.size.pt == 14
+
+    # Проверяем выравнивание через стиль
+    from docx.enum.text import WD_ALIGN_PARAGRAPH
+    assert cap_p.style.paragraph_format.alignment == WD_ALIGN_PARAGRAPH.CENTER
+
+    yaml_path.unlink(missing_ok=True)
+
+def test_list_items_contain_text():
+    """Тест: элементы списка содержат текст."""
+    doc = Document()
+
+    # Добавляем элементы списка
+    p1 = doc.add_paragraph("Первый элемент", style="List Bullet")
+    p2 = doc.add_paragraph("Второй элемент", style="List Number")
+
+    assert p1.text == "Первый элемент"
+    assert p2.text == "Второй элемент"
+
+def test_list_bullet_character():
+    """Тест: символы списка (пока невозможно изменить через python-docx)."""
+    # Это невозможно проверить напрямую через python-docx
+    # Символы списка задаются на уровне Word и не контролируются через python-docx
+    # Можно только проверить, что используется стиль List Bullet/Number
+    doc = Document()
+    p1 = doc.add_paragraph("Элемент", style="List Bullet")
+    p2 = doc.add_paragraph("Элемент", style="List Number")
+
+    assert p1.style.name == "List Bullet"
+    assert p2.style.name == "List Number"
--- a/tests/test_handlers.py
+++ b/tests/test_handlers.py
@@ -87,6 +87,24 @@ def test_handle_list():
    assert doc.paragraphs[0].text == "Элемент 1"
    assert doc.paragraphs[1].text == "Элемент 2"

+def test_handle_list_block_text():
+    doc = Document()
+    style_reg = StyleRegistry(Path("resources/style_config.yaml"))
+    node = {
+        "type": "list",
+        "ordered": False,
+        "children": [
+            {"type": "list_item", "children": [{"type": "block_text", "children": [{"type": "text", "raw": "Элемент 1"}]}]},
+            {"type": "list_item", "children": [{"type": "block_text", "children": [{"type": "text", "raw": "Элемент 2"}]}]},
+        ]
+    }
+    handle_list(node, doc, Path("."), style_reg)
+
+    assert len(doc.paragraphs) == 2
+    assert doc.paragraphs[0].style.name == "List Bullet"
+    assert doc.paragraphs[0].text == "Элемент 1"
+    assert doc.paragraphs[1].text == "Элемент 2"
+

 def test_handle_table():
    doc = Document()
--- a/tests/test_style_registry.py
+++ b/tests/test_style_registry.py
@@ -0,0 +1,87 @@
+import pytest
+from pathlib import Path
+from docx import Document
+from src.core.style_registry import StyleRegistry
+
+def test_apply_style_font_and_color():
+    yaml_content = """
+styles:
+  TestStyle:
+    font:
+      name: "Arial"
+      size: 16
+      bold: true
+      italic: true
+      color: "FF0000"  # красный
+    paragraph:
+      alignment: center
+      space_before: 10
+      space_after: 5
+"""
+    yaml_path = Path("tests/fixtures/test_style.yaml")
+    yaml_path.parent.mkdir(exist_ok=True)
+    yaml_path.write_text(yaml_content, encoding="utf-8")
+
+    doc = Document()
+    style_reg = StyleRegistry(yaml_path)
+    style_reg.ensure_styles_in_doc(doc)
+
+    # Проверим, что стиль создался
+    style = doc.styles["TestStyle"]
+    assert style.font.name == "Arial"
+    assert style.font.size.pt == 16
+    assert style.font.bold is True
+    assert style.font.italic is True
+    assert style.font.color.rgb == (0xFF, 0x00, 0x00)
+
+    # Проверим paragraph_format
+    pf = style.paragraph_format
+    from docx.enum.text import WD_ALIGN_PARAGRAPH
+    assert pf.alignment == WD_ALIGN_PARAGRAPH.CENTER
+    assert pf.space_before.pt == 10
+    assert pf.space_after.pt == 5
+
+    yaml_path.unlink(missing_ok=True)
+
+def test_apply_style_with_defaults():
+    yaml_content = """
+styles:
+  TestStyle:
+    font:
+      name: "Times New Roman"
+      size: 14
+"""
+    yaml_path = Path("tests/fixtures/test_style_defaults.yaml")
+    yaml_path.parent.mkdir(exist_ok=True)
+    yaml_path.write_text(yaml_content, encoding="utf-8")
+
+    doc = Document()
+    style_reg = StyleRegistry(yaml_path)
+    style_reg.ensure_styles_in_doc(doc)
+
+    style = doc.styles["TestStyle"]
+    assert style.font.name == "Times New Roman"
+    assert style.font.size.pt == 14
+    # Проверим, что остальные параметры не изменились (None означает "не задано")
+    assert style.font.bold is None
+    assert style.font.italic is None
+
+    yaml_path.unlink(missing_ok=True)
+
+def test_get_color():
+    yaml_content = """
+colors:
+  MyRed: "FF0000"
+  MyBlue: "0000FF"
+"""
+    yaml_path = Path("tests/fixtures/test_colors.yaml")
+    yaml_path.parent.mkdir(exist_ok=True)
+    yaml_path.write_text(yaml_content, encoding="utf-8")
+
+    style_reg = StyleRegistry(yaml_path)
+    assert style_reg.get_color("MyRed") == (0xFF, 0x00, 0x00)
+    assert style_reg.get_color("MyBlue") == (0x00, 0x00, 0xFF)
+    # Проверим fallback
+    assert style_reg.get_color("NonExistent") == (0x00, 0x00, 0x00)  # чёрный
+
+    yaml_path.unlink(missing_ok=True)