more examples

2025-11-09 21:40:50 +01:00
parent 2b14517344
commit 12ebddaa79
13 changed files with 1906 additions and 3 deletions
@@ -0,0 +1,221 @@
+"""
+Test for pagination example (08_pagination_demo.py).
+
+This test ensures the pagination example runs correctly and produces expected output.
+"""
+
+import pytest
+import sys
+from pathlib import Path
+import importlib.util
+
+# Add examples to path
+examples_dir = Path(__file__).parent.parent.parent / "examples"
+sys.path.insert(0, str(examples_dir))
+
+# Load the demo module
+spec = importlib.util.spec_from_file_location(
+    "pagination_demo",
+    examples_dir / "08_pagination_demo.py"
+)
+demo_module = importlib.util.module_from_spec(spec)
+spec.loader.exec_module(demo_module)
+
+
+def test_pagination_demo_imports():
+    """Test that the pagination demo can be imported without errors."""
+    assert demo_module is not None
+
+
+def test_create_sample_paragraph():
+    """Test creating a sample paragraph."""
+    create_sample_paragraph = demo_module.create_sample_paragraph
+    from pyWebLayout.abstract.block import Paragraph
+
+    para = create_sample_paragraph("Hello world test")
+    assert isinstance(para, Paragraph)
+    assert len(para.words) == 3
+    assert para.words[0].text == "Hello"
+    assert para.words[1].text == "world"
+    assert para.words[2].text == "test"
+
+
+def test_create_title_paragraph():
+    """Test creating a title paragraph."""
+    create_title_paragraph = demo_module.create_title_paragraph
+    from pyWebLayout.abstract.block import Paragraph
+
+    title = create_title_paragraph("Test Title")
+    assert isinstance(title, Paragraph)
+    assert len(title.words) == 2
+    # Title should have larger font
+    assert title.style.font_size == 24
+
+
+def test_create_heading_paragraph():
+    """Test creating a heading paragraph."""
+    create_heading_paragraph = demo_module.create_heading_paragraph
+    from pyWebLayout.abstract.block import Paragraph
+
+    heading = create_heading_paragraph("Test Heading")
+    assert isinstance(heading, Paragraph)
+    assert len(heading.words) == 2
+    # Heading should have medium font
+    assert heading.style.font_size == 18
+
+
+def test_create_placeholder_image():
+    """Test creating a placeholder image."""
+    create_placeholder_image = demo_module.create_placeholder_image
+    from pyWebLayout.abstract.block import Image as AbstractImage
+
+    img = create_placeholder_image(200, 150, "Test Image")
+    assert isinstance(img, AbstractImage)
+    assert img.source.size == (200, 150)
+
+
+def test_pagebreak_layouter_integration():
+    """Test that PageBreak properly forces new pages."""
+    create_sample_paragraph = demo_module.create_sample_paragraph
+    from pyWebLayout.abstract.block import PageBreak
+    from pyWebLayout.concrete.page import Page
+    from pyWebLayout.style.page_style import PageStyle
+    from pyWebLayout.layout.document_layouter import DocumentLayouter
+
+    # Create a small page
+    page_style = PageStyle(padding=(10, 10, 10, 10))
+    page1 = Page(size=(200, 200), style=page_style)
+    layouter1 = DocumentLayouter(page1)
+
+    # Add some content
+    para1 = create_sample_paragraph("First paragraph")
+    success, _, _ = layouter1.layout_paragraph(para1)
+    assert success
+
+    # Record y offset before pagebreak
+    y_before_break = page1._current_y_offset
+
+    # Simulating PageBreak by creating new page
+    # (PageBreak doesn't get laid out - it signals page creation)
+    page2 = Page(size=(200, 200), style=page_style)
+    layouter2 = DocumentLayouter(page2)
+
+    # Verify new page starts at initial offset
+    initial_offset = page2._current_y_offset
+
+    # Add content to new page
+    para2 = create_sample_paragraph("Second paragraph")
+    success, _, _ = layouter2.layout_paragraph(para2)
+    assert success
+
+    # New page should start fresh (at border_size + padding)
+    # Both pages should have same initial offset
+    assert initial_offset == page1.border_size + page_style.padding_top
+
+
+def test_example_document_with_pagebreaks():
+    """Test creating the main example document with pagebreaks."""
+    create_example_document_with_pagebreaks = demo_module.create_example_document_with_pagebreaks
+
+    pages = create_example_document_with_pagebreaks()
+
+    # Should create multiple pages
+    assert len(pages) > 1
+    # Should have created 5 pages (based on PageBreak placements)
+    assert len(pages) == 5
+
+    # Each page should be valid
+    for page in pages:
+        assert page.size == (600, 800)
+        # Page should have some content or be a clean break page
+        assert page._current_y_offset >= page.border_size
+
+
+def test_auto_pagination_example():
+    """Test auto-pagination without explicit PageBreaks."""
+    create_auto_pagination_example = demo_module.create_auto_pagination_example
+
+    pages = create_auto_pagination_example()
+
+    # Should create multiple pages due to content overflow
+    assert len(pages) >= 1
+
+    # Each page should be valid
+    for page in pages:
+        assert page.size == (500, 600)
+
+
+def test_add_page_numbers():
+    """Test adding page numbers to rendered pages."""
+    add_page_numbers = demo_module.add_page_numbers
+    from pyWebLayout.concrete.page import Page
+    from pyWebLayout.style.page_style import PageStyle
+
+    # Create a few simple pages
+    page_style = PageStyle()
+    pages = [
+        Page(size=(200, 200), style=page_style),
+        Page(size=(200, 200), style=page_style),
+        Page(size=(200, 200), style=page_style)
+    ]
+
+    # Add page numbers
+    numbered = add_page_numbers(pages, start_number=1)
+
+    # Should return same number of pages
+    assert len(numbered) == 3
+
+    # Each should be a rendered image
+    from PIL import Image
+    for img in numbered:
+        assert isinstance(img, Image.Image)
+        assert img.size == (200, 200)
+
+
+def test_combine_pages_vertically():
+    """Test combining pages into vertical strip."""
+    combine_pages_vertically = demo_module.combine_pages_vertically
+    from pyWebLayout.concrete.page import Page
+    from pyWebLayout.style.page_style import PageStyle
+
+    # Create a few pages
+    page_style = PageStyle()
+    pages = [
+        Page(size=(200, 200), style=page_style).render(),
+        Page(size=(200, 200), style=page_style).render()
+    ]
+
+    combined = combine_pages_vertically(pages, title="Test Title")
+
+    # Should create a combined image
+    from PIL import Image
+    assert isinstance(combined, Image.Image)
+
+    # Should be taller than individual pages
+    assert combined.size[1] > 200
+
+
+def test_main_function():
+    """Test that the main function runs without errors."""
+    main = demo_module.main
+
+    # Run main (will create output files)
+    result = main()
+
+    # Should return two combined images
+    assert result is not None
+    assert len(result) == 2
+
+    # Both should be PIL images
+    from PIL import Image
+    assert isinstance(result[0], Image.Image)
+    assert isinstance(result[1], Image.Image)
+
+    # Output files should exist
+    output_dir = Path("docs/images")
+    assert (output_dir / "example_08_pagination_explicit.png").exists()
+    assert (output_dir / "example_08_pagination_auto.png").exists()
+
+
+if __name__ == "__main__":
+    pytest.main([__file__, "-v"])