diff --git a/tests/test_sentence_reordering.py b/tests/test_sentence_reordering.py index 9ed5b4da..05645c70 100644 --- a/tests/test_sentence_reordering.py +++ b/tests/test_sentence_reordering.py @@ -37,6 +37,7 @@ def test_getitem(dataset, config): assert "metadata" in item assert item["metadata"]["word_count"] >= config.min_words_in_sentence assert item["metadata"]["word_count"] <= config.max_words_in_sentence + assert len(item['answer'].split()) == item['metadata']['word_count'] def test_key_error_in_getitem(dataset):