ADD TEST

wuyunxiangwyx · Dec 1, 2020 · 4f054ca · 4f054ca
1 parent 359f17a
commit 4f054ca
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/test_span.py b/tests/test_span.py
@@ -0,0 +1,6 @@
+from ltp.utils import get_entities
+
+
+def test_span():
+    span = ['B-PER', 'I-PER', 'O', 'B-LOC']
+    assert get_entities(span) == [('PER', 0, 1), ('LOC', 3, 3)]