feat: pdf2cad

tests/test_geometry_extractor.py

@@ -0,0 +1,74 @@
+"""Tests for PDF vector geometry extraction."""
+import pytest
+import pymupdf
+from pathlib import Path
+
+from pdf2imos.extract.geometry import extract_geometry
+from pdf2imos.models import PageExtraction, RawPath
+
+FIXTURES_DIR = Path(__file__).parent / "fixtures" / "input"
+
+
+class TestExtractGeometry:
+    def test_returns_page_extraction(self, simple_panel_pdf):
+        doc = pymupdf.open(str(simple_panel_pdf))
+        result = extract_geometry(doc[0])
+        assert isinstance(result, PageExtraction)
+
+    def test_paths_are_raw_path_objects(self, simple_panel_pdf):
+        doc = pymupdf.open(str(simple_panel_pdf))
+        result = extract_geometry(doc[0])
+        assert all(isinstance(p, RawPath) for p in result.paths)
+
+    def test_extracts_sufficient_paths(self, simple_panel_pdf):
+        """simple_panel.pdf should have >10 paths."""
+        doc = pymupdf.open(str(simple_panel_pdf))
+        result = extract_geometry(doc[0])
+        assert len(result.paths) > 10, f"Expected >10 paths, got {len(result.paths)}"
+
+    def test_dashes_extracted_correctly(self, simple_panel_pdf):
+        """Solid lines have empty dashes, dashed lines have non-empty dashes."""
+        doc = pymupdf.open(str(simple_panel_pdf))
+        result = extract_geometry(doc[0])
+        solid = [p for p in result.paths if not p.dashes]
+        # Should have at least some solid lines (geometry outline)
+        assert len(solid) > 0, "No solid lines found"
+
+    def test_y_coordinates_flipped(self, simple_panel_pdf):
+        """After y-flip, rect y0 should be >= 0 and <= page_height."""
+        doc = pymupdf.open(str(simple_panel_pdf))
+        page = doc[0]
+        result = extract_geometry(page)
+        page_h = result.page_height
+        for p in result.paths:
+            x0, y0, x1, y1 = p.rect
+            assert y0 >= -0.1, f"y0 negative: {y0}"
+            assert y1 <= page_h + 0.1, f"y1 > page_height: {y1}"
+
+    def test_texts_empty_in_result(self, simple_panel_pdf):
+        """extract_geometry returns empty texts (text extracted separately)."""
+        doc = pymupdf.open(str(simple_panel_pdf))
+        result = extract_geometry(doc[0])
+        assert result.texts == (), "extract_geometry should return empty texts"
+
+    def test_page_dimensions_stored(self, simple_panel_pdf):
+        """Page width and height stored correctly."""
+        doc = pymupdf.open(str(simple_panel_pdf))
+        page = doc[0]
+        result = extract_geometry(page)
+        assert result.page_width == pytest.approx(page.rect.width)
+        assert result.page_height == pytest.approx(page.rect.height)
+
+    def test_all_fixtures_extractable(self, all_fixture_pdfs):
+        """All fixture PDFs can be extracted without error."""
+        for pdf_path in all_fixture_pdfs:
+            doc = pymupdf.open(str(pdf_path))
+            result = extract_geometry(doc[0])
+            assert len(result.paths) > 0, f"No paths in {pdf_path.name}"
+
+    def test_width_stored_in_rawpath(self, simple_panel_pdf):
+        """RawPath.width field populated."""
+        doc = pymupdf.open(str(simple_panel_pdf))
+        result = extract_geometry(doc[0])
+        widths = {p.width for p in result.paths}
+        assert len(widths) > 1, "Expected multiple distinct line widths"