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)