change ExactAnalyzer to SubformAnalyzer

KokocGroup · Dec 23, 2015 · fa96ab8 · fa96ab8
1 parent d5799dc
commit fa96ab8
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/ta4/__init__.py b/ta4/__init__.py
@@ -178,7 +178,7 @@ def absorptions(phrases):
 
 
 def is_contains(haystack, needle):
-    return bool(ExactAnalyzer().mark(needle, haystack, 0, should_mark=False))
+    return bool(SubformsAnalyzer().mark(needle, haystack, 0, should_mark=False))
 
 
 def activate_marker(marker_sentence, marker, sentence):

diff --git a/tests/test_absorptions.py b/tests/test_absorptions.py
@@ -58,6 +58,21 @@
          (u"[подъёмник]", 0),
          (u"[мачтовый]", 0)]
     ),
+    (
+        [(u'мачтовые подъемники', 1),
+         (u'[мачтовый]', 2),
+         (u'мачтовые', 1),
+         (u'подъемники', 1),
+         (u'[подъёмник]', 2),
+         (u'[мачтовый] [подъёмник]', 2)],
+
+        [(u'мачтовые подъемники', 1),
+         (u'[мачтовый]', 0),
+         (u'мачтовые', 0),
+         (u'подъемники', 0),
+         (u'[подъёмник]', 0),
+         (u'[мачтовый] [подъёмник]', 1)],
+    )
 ])
 def test_absorptions(input, result):
     assert set(absorptions(input)) == set(result)