from gfa_core.mas.utils.intent_parser import (
    parse_explicit_species_role_mentions,
    parse_gene_family_scope_selection,
)


def test_ref_data_path_does_not_assign_species_roles() -> None:
    query = (
        "苦荞基因组数据在 "
        "/home/xia-lab/wutx_old/wutx/plantAnalysisAgent/ref_data/Tartary_genome，"
        "拟南芥基因组数据在 "
        "/home/xia-lab/wutx_old/wutx/plantAnalysisAgent/ref_data/ath，"
        "Ft是目标物种，At是参考物种"
    )

    scope = parse_gene_family_scope_selection(query)

    assert scope["targetSpecies"] == ["Ft"]
    assert scope["referenceSpecies"] == ["At"]


def test_explicit_role_parser_keeps_unassigned_mentions_separate() -> None:
    roles = parse_explicit_species_role_mentions(
        "Os和At的数据已经准备好；Os是目标物种，At是参考物种，Ppy稍后再分析"
    )

    assert roles == {
        "targetSpecies": ["Os"],
        "referenceSpecies": ["At"],
        "comparativeSpecies": [],
        "unassigned": ["Ppy"],
    }