You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
47 lines
1.2 KiB
47 lines
1.2 KiB
6 years ago
|
#!/usr/bin/env python
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
# pylint: disable=no-self-use, pointless-statement, missing-docstring, invalid-name, pointless-string-statement
|
||
|
|
|
||
|
|
from rebulk.match import Matches, Match
|
||
|
|
|
||
|
|
from ...rules.processors import StripSeparators
|
||
|
|
|
||
|
|
|
||
|
|
def test_strip_separators():
|
||
|
|
strip_separators = StripSeparators()
|
||
|
|
|
||
|
|
matches = Matches()
|
||
|
|
|
||
|
|
m = Match(3, 11, input_string="pre.ABCDEF.post")
|
||
|
|
|
||
|
|
assert m.raw == '.ABCDEF.'
|
||
|
|
matches.append(m)
|
||
|
|
|
||
|
|
returned_matches = strip_separators.when(matches, None)
|
||
|
|
assert returned_matches == matches
|
||
|
|
|
||
|
|
strip_separators.then(matches, returned_matches, None)
|
||
|
|
|
||
|
|
assert m.raw == 'ABCDEF'
|
||
|
|
|
||
|
|
|
||
|
|
def test_strip_separators_keep_acronyms():
|
||
|
|
strip_separators = StripSeparators()
|
||
|
|
|
||
|
|
matches = Matches()
|
||
|
|
|
||
|
|
m = Match(0, 13, input_string=".S.H.I.E.L.D.")
|
||
|
|
m2 = Match(0, 22, input_string=".Agent.Of.S.H.I.E.L.D.")
|
||
|
|
|
||
|
|
assert m.raw == '.S.H.I.E.L.D.'
|
||
|
|
matches.append(m)
|
||
|
|
matches.append(m2)
|
||
|
|
|
||
|
|
returned_matches = strip_separators.when(matches, None)
|
||
|
|
assert returned_matches == matches
|
||
|
|
|
||
|
|
strip_separators.then(matches, returned_matches, None)
|
||
|
|
|
||
|
|
assert m.raw == '.S.H.I.E.L.D.'
|
||
|
|
assert m2.raw == 'Agent.Of.S.H.I.E.L.D.'
|