12 lines
493 B
Python
12 lines
493 B
Python
from tools.graphrag.graphrag import extract_works
|
|
from .test_graphrag import make_document, make_chunk
|
|
|
|
|
|
def test_scientific_project_heading_has_research_project_semantics():
|
|
works = extract_works(make_document(categories=['vp2024']), [make_chunk(
|
|
0, ['Vedecký projekt 2024'],
|
|
'Sekcia: Vedecký projekt 2024\nNázov: Overovanie jazykových modelov\n')])
|
|
assert len(works) == 1
|
|
assert works[0]['work_type'] == 'research_project'
|
|
assert works[0]['year'] == 2024
|