def test_simple_string_matcher():
values = ["China", "Beijing", "City"]
string_matcher = StringMatcher()
string_matcher.init(values)
for value in values:
match = string_matcher.find(value)
if match is not None:
assert value == match[0].text
def test_simple_with_ids_string_matcher():
values = ["China", "Beijing", "City"]
ids = ["1", "2", "3"]
string_matcher = StringMatcher()
string_matcher.init(values, ids)
for i in range(0, len(values)):
value = values[i]
match = string_matcher.find(value)
if match is not None:
assert value == match[0].text
assert ids[i] == match[0].canonical_values[0]
def test_string_matcher():
utc_8_value = 'UTC+08:00'
utc_8_words = ['beijing time', 'chongqing time', 'hong kong time', 'urumqi time']
utc_2_value = 'UTC+02:00'
utc_2_words = ['cairo time', 'beirut time', 'gaza time', 'amman time']
value_dictionary = {utc_2_value: utc_2_words, utc_8_value: utc_8_words}
string_matcher = StringMatcher()
string_matcher.init(value_dictionary)
for value in utc_8_words:
sentence = 'please change {}, thanks'.format(value)
matches: [MatchResult] = string_matcher.find(sentence)
assert value == matches[0].text
assert utc_8_value == matches[0].canonical_values[0]
assert 14 == matches[0].start
for value in utc_2_words:
sentence = 'please change {}, thanks'.format(value)
matches: [MatchResult] = string_matcher.find(sentence)
assert value == matches[0].text
assert str(utc_2_value) == matches[0].canonical_values[0]
assert 14 == matches[0].start