Update temporal test for thyme2 model.

Machine-Learning-for-Medical-Language · Jun 27, 2023 · 2613560 · 2613560
1 parent 100cb52
commit 2613560
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/test/test_api.py b/test/test_api.py
@@ -51,11 +51,16 @@ def test_temporal_process_sentence(self, startup_temporal):
                          {'begin': 5, 'dtr': 'BEFORE', 'end': 5},
                          {'begin': 13, 'dtr': 'AFTER', 'end': 13},
                          {'begin': 15, 'dtr': 'AFTER', 'end': 15}]],
-             'relations': [[{'arg1': 'EVENT-0', 'arg2': 'EVENT-1', 'category': 'OVERLAP', 'arg1_start': 3, 'arg2_start': 5}, 
-                            {'arg1': 'TIMEX-0', 'arg2': 'EVENT-0', 'category': 'CONTAINS', 'arg1_start': 6, 'arg2_start': 3}, 
-                            {'arg1': 'EVENT-3', 'arg2': 'EVENT-2', 'category': 'BEGINS-ON', 'arg1_start': 15, 'arg2_start': 13}, 
-                            {'arg1': 'TIMEX-1', 'arg2': 'EVENT-2', 'category': 'CONTAINS', 'arg1_start': 16, 'arg2_start': 13}, 
-                            {'arg1': 'TIMEX-1', 'arg2': 'EVENT-3', 'category': 'CONTAINS', 'arg1_start': 16, 'arg2_start': 15}]],
+             'relations': [[{'arg1': 'TIMEX-0',
+                 'arg1_start': 6,
+                 'arg2': 'EVENT-0',
+                 'arg2_start': 3,
+                 'category': 'CONTAINS'},
+                {'arg1': 'TIMEX-1',
+                 'arg1_start': 16,
+                 'arg2': 'EVENT-2',
+                 'arg2_start': 13,
+                 'category': 'CONTAINS'}]],
              'timexes': [[{'begin': 6, 'end': 9, 'timeClass': 'DATE'},
                           {'begin': 16, 'end': 17, 'timeClass': 'DATE'}]]}
         )