Fix unit tests.

ckiplab · Apr 24, 2020 · 600a9c1 · 600a9c1
1 parent 05c354d
commit 600a9c1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/container/tree_parsed.py b/test/container/tree_parsed.py
@@ -287,7 +287,7 @@ def test_get_relations_semantic(self):
             (21, 22, 'aspect',),
         }
         rels_id_out = {
-            (rel.head.identifier, rel.tail.identifier, rel.relation)
+            (rel.head.identifier, rel.tail.identifier, rel.relation.data.role)
                 for rel in obj.get_relations(semantic=True)
         }
         self.assertEqual(rels_id_out, rels_id)
@@ -310,7 +310,7 @@ def test_get_relations_syntactic(self):
             (21, 22, 'aspect',),
         }
         rels_id_out = {
-            (rel.head.identifier, rel.tail.identifier, rel.relation)
+            (rel.head.identifier, rel.tail.identifier, rel.relation.data.role)
                 for rel in obj.get_relations(semantic=False)
         }
         self.assertEqual(rels_id_out, rels_id)