From 28d164beaca050ff1ed3bc41eac09964c790fbad Mon Sep 17 00:00:00 2001 From: Arne Binder Date: Thu, 2 Nov 2023 18:59:37 +0100 Subject: [PATCH] fix test --- tests/dataset_builders/pie/test_conll2003.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/dataset_builders/pie/test_conll2003.py b/tests/dataset_builders/pie/test_conll2003.py index 9bb5e8f1..017eb8b8 100644 --- a/tests/dataset_builders/pie/test_conll2003.py +++ b/tests/dataset_builders/pie/test_conll2003.py @@ -106,10 +106,10 @@ def converted_document(converted_pie_dataset): def test_converted_document(converted_document, converter_document_type): - assert isinstance(document, converter_document_type) + assert isinstance(converted_document, converter_document_type) if converter_document_type == TextDocumentWithLabeledSpans: - assert document.text == "EU rejects German call to boycott British lamb ." - entities = list(document.labeled_spans) + assert converted_document.text == "EU rejects German call to boycott British lamb ." + entities = list(converted_document.labeled_spans) assert len(entities) == 3 assert str(entities[0]) == "EU" assert str(entities[1]) == "German"