From b2173f3ef09942226d950db221c4b068393bb4ed Mon Sep 17 00:00:00 2001
From: Vishesh 'ironeagle' Bangotra <aetoskia@gmail.com>
Date: Fri, 2 Jan 2026 19:02:20 +0530
Subject: [PATCH] refactor(tests): use omniread public API instead of internal
 module imports

- Replace deep imports with top-level omniread exports in tests
- Ensure tests validate only the supported public API surface
- Align HTML and PDF tests with documented library usage
---
 tests/conftest.py         | 15 +++++++++++----
 tests/test_html_simple.py |  9 +++++++--
 tests/test_html_table.py  |  9 +++++++--
 tests/test_pdf_simple.py  |  8 ++++++--
 4 files changed, 31 insertions(+), 10 deletions(-)

diff --git a/tests/conftest.py b/tests/conftest.py
index 23d7425..309c278 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -4,10 +4,17 @@ import httpx
 from pathlib import Path
 from jinja2 import Environment, BaseLoader
 
-from omniread.core.content import ContentType
-from omniread.html.scraper import HTMLScraper
-from omniread.pdf.client import FileSystemPDFClient
-from omniread.pdf.scraper import PDFScraper
+from omniread import (
+    # core
+    ContentType,
+
+    # html
+    HTMLScraper,
+
+    # pdf
+    FileSystemPDFClient,
+    PDFScraper,
+)
 
 
 MOCK_HTML_DIR = Path(__file__).parent / "mocks" / "html"
diff --git a/tests/test_html_simple.py b/tests/test_html_simple.py
index c5129d2..cd32f0c 100644
--- a/tests/test_html_simple.py
+++ b/tests/test_html_simple.py
@@ -3,8 +3,13 @@ from typing import Optional
 from pydantic import BaseModel
 from bs4 import Tag
 
-from omniread.html.parser import HTMLParser
-from omniread.core.content import Content
+from omniread import (
+    # core
+    Content,
+
+    # html
+    HTMLParser,
+)
 
 
 class ParsedSimpleHTML(BaseModel):
diff --git a/tests/test_html_table.py b/tests/test_html_table.py
index e7b998a..dd4a133 100644
--- a/tests/test_html_table.py
+++ b/tests/test_html_table.py
@@ -2,8 +2,13 @@ from typing import Optional
 
 from pydantic import BaseModel
 
-from omniread.html.parser import HTMLParser
-from omniread.core.content import Content
+from omniread import (
+    # core
+    Content,
+
+    # html
+    HTMLParser,
+)
 
 
 class ParsedTableHTML(BaseModel):
diff --git a/tests/test_pdf_simple.py b/tests/test_pdf_simple.py
index 552c8a5..b90f791 100644
--- a/tests/test_pdf_simple.py
+++ b/tests/test_pdf_simple.py
@@ -1,9 +1,13 @@
 from typing import Literal
 from pydantic import BaseModel
 
-from omniread.pdf import PDFParser
-from omniread.core.content import Content
+from omniread import (
+    # core
+    Content,
 
+    # pdf
+    PDFParser,
+)
 
 class ParsedPDF(BaseModel):
     size_bytes: int