"""Tests para build_relation_schema_prompt."""

from build_relation_schema_prompt import build_relation_schema_prompt


def test_lista_normal():
    relation_types = ["funds", "employs", "communicates_with", "owns", "operates"]
    result = build_relation_schema_prompt(relation_types)
    assert result.startswith("Allowed relation types:")
    assert "funds" in result
    assert "employs" in result
    assert "communicates_with" in result
    assert "owns" in result
    assert "operates" in result


def test_lista_vacia_retorna_string_vacio():
    result = build_relation_schema_prompt([])
    assert result == ""
