48 lines
1.4 KiB
Python
Executable File
48 lines
1.4 KiB
Python
Executable File
from doclift.legacy_doc import FigureAsset, extract_references, extract_tables, link_related_assets
|
|
|
|
|
|
def test_extract_references_dedupes() -> None:
|
|
refs = extract_references("See Table 1 and table 1 and Table 2.", r"\bTable\s+\d+\b")
|
|
assert refs == ["Table 1", "Table 2"]
|
|
|
|
|
|
def test_extract_tables_parses_tabbed_rows() -> None:
|
|
text = "\n".join(
|
|
[
|
|
"Intro",
|
|
"Table 1. Example caption",
|
|
"",
|
|
"Metric\tRest\tSwim",
|
|
"O2\t1.0\t2.0",
|
|
"CO2\t0.5\t1.1",
|
|
]
|
|
)
|
|
tables = extract_tables(text)
|
|
assert len(tables) == 1
|
|
assert tables[0].caption == "Table 1. Example caption"
|
|
assert tables[0].column_count_guess == 3
|
|
assert tables[0].parsed_rows[1] == ["O2", "1.0", "2.0"]
|
|
|
|
|
|
def test_link_related_assets_matches_explicit_figure_refs() -> None:
|
|
assets = [
|
|
FigureAsset(
|
|
asset_id="a1",
|
|
path="/tmp/Fig. 5.1.bmp",
|
|
relative_path="vol/Fig. 5.1.bmp",
|
|
name="Fig. 5.1.bmp",
|
|
container="vol",
|
|
looks_like_figure=True,
|
|
),
|
|
FigureAsset(
|
|
asset_id="a2",
|
|
path="/tmp/Slide 1.jpg",
|
|
relative_path="vol/Slide 1.jpg",
|
|
name="Slide 1.jpg",
|
|
container="vol",
|
|
looks_like_figure=False,
|
|
),
|
|
]
|
|
matched = link_related_assets(["Fig. 5.1"], assets)
|
|
assert [asset.asset_id for asset in matched] == ["a1"]
|