add test

tamuhey · Mar 3, 2021 · c9dcca4 · c9dcca4
1 parent 679336e
commit c9dcca4
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/python/tests/test_main.py b/python/tests/test_main.py
@@ -46,7 +46,10 @@ def test_random_get_original_spans2(tokens, text, expected):
 
 @pytest.mark.parametrize(
     "span, spans, expected",
-    [((3, 10), [(0, 3), (3, 4), (4, 9), (9, 12)], ((1, True), (4, False)))],
+    [
+        ((3, 10), [(0, 3), (3, 4), (4, 9), (9, 12)], ((1, True), (4, False))),
+        ((0, 18), [(0, 13), (13, 18)], ((0, True), (2, True))),
+    ],
 )
 def test_lift_span_index(span, spans, expected):
     assert textspan.lift_span_index(span, spans) == expected