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)