52 lines
1.6 KiB
Python
52 lines
1.6 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
from afs_scawful.research import build_research_catalog, extract_abstract_excerpt
|
|
|
|
|
|
def test_extract_abstract_excerpt() -> None:
|
|
text = "Title\nAbstract\nThis is the abstract.\n1 Introduction\nBody"
|
|
assert extract_abstract_excerpt(text, max_chars=200) == "This is the abstract."
|
|
|
|
|
|
def test_build_research_catalog_regex(tmp_path: Path) -> None:
|
|
research_root = tmp_path / "Research"
|
|
research_root.mkdir()
|
|
pdf_path = research_root / "paper.pdf"
|
|
pdf_path.write_bytes(
|
|
b"not a real pdf /Title (Test Paper) /Author (Jane Doe)",
|
|
)
|
|
|
|
catalog = build_research_catalog(research_root, include_abstract=False)
|
|
assert catalog["count"] == 1
|
|
entry = catalog["papers"][0]
|
|
assert entry["title"] == "Test Paper"
|
|
assert entry["author"] == "Jane Doe"
|
|
assert entry["metadata_source"] == "regex"
|
|
|
|
|
|
def test_build_research_catalog_overrides(tmp_path: Path) -> None:
|
|
research_root = tmp_path / "Research"
|
|
research_root.mkdir()
|
|
pdf_path = research_root / "paper.pdf"
|
|
pdf_path.write_bytes(b"%PDF-1.0 /Title (Ignored)")
|
|
|
|
overrides = {
|
|
"papers": {
|
|
"paper.pdf": {
|
|
"title": "Manual Title",
|
|
"author": "Manual Author",
|
|
}
|
|
}
|
|
}
|
|
catalog = build_research_catalog(
|
|
research_root,
|
|
overrides=overrides,
|
|
include_abstract=False,
|
|
)
|
|
entry = catalog["papers"][0]
|
|
assert entry["title"] == "Manual Title"
|
|
assert entry["author"] == "Manual Author"
|
|
assert entry["metadata_source"] == "override"
|