News-Search-Engine / tests /conftest.py
1xMezoo's picture
Deploy News Search Engine to Hugging Face Spaces
39ff835
Raw
History Blame Contribute Delete
433 Bytes
"""Shared pytest fixtures."""
from pathlib import Path
import pytest
from news_search import SearchEngine, build_index, load_corpus
SAMPLE = Path(__file__).resolve().parents[1] / "data" / "sample_news.jsonl"
@pytest.fixture(scope="session")
def engine() -> SearchEngine:
"""A SearchEngine over the committed sample corpus (built once per session)."""
return SearchEngine(build_index(load_corpus(SAMPLE), verbose=False))