Fix test

huji-nlp · Apr 16, 2019 · 5cd08c5 · 5cd08c5
1 parent d7f13f5
commit 5cd08c5
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/ucca/tests/test_textutil.py b/ucca/tests/test_textutil.py
@@ -85,9 +85,11 @@ def test_preannotate_passage(create, as_array, convert_and_back, partial, monkey
                     terminal.extra[attr.key] = value
     passage = (passage, convert.from_standard(convert.to_standard(passage)))[convert_and_back]
     if not partial:
-        assert textutil.is_annotated(passage, as_array=as_array), "Passage %s is not pre-annotated" % passage.ID
-    textutil.annotate(passage, as_array=as_array)
-    assert textutil.is_annotated(passage, as_array=as_array), "Passage %s is not annotated" % passage.ID
+        assert textutil.is_annotated(passage, as_array=as_array, as_extra=not as_array), \
+            "Passage %s is not pre-annotated" % passage.ID
+    textutil.annotate(passage, as_array=as_array, as_extra=not as_array)
+    assert textutil.is_annotated(passage, as_array=as_array, as_extra=not as_array), \
+        "Passage %s is not annotated" % passage.ID
     for terminal in l0.all:
         for i, (attr, value) in enumerate(zip(textutil.Attr, attr_values)):
             if value: