def test_names():
auto = Automaton()
auto.add_all(NAMES)
auto.update_automaton()
auto_matches = [(m.start, m.end) for m in auto.get_matches(TEXT)]
with TemporaryDirectory() as tmpdir:
#tmpdir = ''
fnm = os.path.join(tmpdir, 'test.aca')
auto.save_to_file(fnm)
auto2 = Automaton()
auto2.load_from_file(fnm)
auto2_matches = [(m.start, m.end) for m in auto2.get_matches(TEXT)]
assert list(auto.items()) == list(auto2.items())
assert list(auto.prefixes()) == list(auto2.prefixes())
assert auto_matches == auto2_matches
auto3 = Automaton()
auto3.load_from_string(auto2.save_to_string())
auto3_matches = [(m.start, m.end) for m in auto3.get_matches(TEXT)]
assert list(auto.items()) == list(auto2.items())
assert list(auto.prefixes()) == list(auto2.prefixes())
assert auto_matches == auto3_matches
def test_names():
auto = Automaton()
auto[KEY] = VAL
auto.update_automaton()
with TemporaryDirectory() as tmpdir:
fnm = os.path.join(tmpdir, 'test.aca')
auto.save_to_file(fnm)
auto2 = Automaton()
auto2.load_from_file(fnm)
assert auto2[KEY] == VAL
# Import the library and initiate the automaton
from aca import Automaton
automaton = Automaton()
# add the entities and build the automaton
automaton.add_all(['Funderbeam', 'Funderbeam Data', 'Funderbeam Markets'])
automaton.update_automaton()
# find matches
text = 'Funderbeam Data and Funderbeam Markets are two different products of Funderbeam'
for match in automaton.get_matches(text, exclude_overlaps=False):
print(match.start, match.end, match.elems)
for match in automaton.get_matches(text, exclude_overlaps=True):
print(match.start, match.end, match.elems)
