dp-zp-agent/tools/graphrag/tests/test_title_labels.py

17 lines
674 B
Python

import pytest
from tools.graphrag.graphrag import extract_work_title
from tools.graphrag.evaluation.source_oracle import acceptable_title
@pytest.mark.parametrize("label", ["rešerš", "tutoriál", "projekt", "dokument", "repozitár"])
def test_generic_deliverable_is_not_an_academic_title(label):
assert extract_work_title([f"- Písomná práca: [{label}](./report)"]) is None
assert not acceptable_title(label, explicit=True)
def test_specific_deliverable_title_remains_supported():
title = "Rešerš na tému prekladu"
assert extract_work_title([f"- Písomná práca: [{title}](./report)"]) == title
assert acceptable_title(title, explicit=True)