| from src.model.document import InputDoc | |
| short_example = "short example" | |
| title_example = "!!this is about Aldo Moro" | |
| def test_short_example(): | |
| text = short_example | |
| assert InputDoc(fulltext=text).container.normals[0].text == text | |
| def test_title_example(): | |
| text = title_example | |
| doc = InputDoc(fulltext=text) | |
| assert doc.title == text.split('!!')[1] | |
| assert doc.container.title == title_example.split('!!')[1] | |
| assert not doc.container.children | |
| def test_long_example(): | |
| text = open("../data/long_example.txt", "r").read() | |
| doc = InputDoc(fulltext=text) | |
| tasks = doc.tasks | |
| assert doc.title == "terrorism in Italy in the years 70 and 80" | |
| assert doc.container.title == "terrorism in Italy in the years 70 and 80" | |
| assert len(doc.container.children) == 2 | |