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"]