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