def test_det_search_ab_aa(self):
f = MutableFST()
s0 = f.get_initial_state()
s1 = f.add_state()
s2 = f.add_state()
s3 = f.add_state()
f.add_transition(s0, "a", s1)
f.add_transition(s1, "a", s2).add_transition(s1, "b", s3)
f.set_final_state(s2).set_final_state(s3)
f.add_output(s2, "aa").add_output(s3, "ab")
self.assertEqual(f.det_search("a ab aa"), [])
def test_det_search_ab_babb_bb(self):
# build some fsa
f = MutableFST()
s0 = f.get_initial_state()
s1 = f.add_state()
s2 = f.add_state()
s3 = f.add_state()
s4 = f.add_state()
s5 = f.add_state()
s6 = f.add_state()
s7 = f.add_state()
f.add_transition(s0, "a", s1).add_transition(s0, "b", s3)
f.add_transition(s1, "a", s1).add_transition(s1, "b", s2)
f.add_transition(s2, "a", s4).add_transition(s2, "b", s7)
f.add_transition(s3, "a", s4).add_transition(s3, "b", s7)
f.add_transition(s4, "a", s1).add_transition(s4, "b", s5)
f.add_transition(s5, "a", s4).add_transition(s5, "b", s6)
f.add_transition(s6, "a", s4).add_transition(s6, "b", s7)
f.add_transition(s7, "a", s4).add_transition(s7, "b", s7)
f.set_final_state(s2).set_final_state(s5).set_final_state(s6).set_final_state(s7)
f.add_output(s2, "ab").add_output(s5, "ab").add_output(s6, "bb").add_output(s6, "babb").add_output(s7, "bb")
self.assertEqual(
f.to_dict(),
{
"initial": s0,
"finals": [s2, s5, s6, s7],
"outputs": {s2: ["ab"], s5: ["ab"], s6: ["babb", "bb"], s7: ["bb"]},
"transitions": {
s0: {"a": s1, "b": s3},
s1: {"a": s1, "b": s2},
s2: {"a": s4, "b": s7},
s3: {"a": s4, "b": s7},
s4: {"a": s1, "b": s5},
s5: {"a": s4, "b": s6},
s6: {"a": s4, "b": s7},
s7: {"a": s4, "b": s7},
},
},
)
self.assertEqual(f.get_letters(), ["a", "b"])
# test det_search simulation on babba
self.assertEqual(f.get_target(s0, "b"), s3)
self.assertEqual(f.get_target(s3, "a"), s4)
self.assertEqual(f.get_target(s4, "b"), s5)
self.assertEqual(f.get_target(s5, "b"), s6)
self.assertEqual(f.get_target(s6, "a"), s4)
# test det_search
outputs = f.det_search("babba")
# babba
# 01234
self.assertEqual(outputs, [("ab", 1, 3), ("babb", 0, 4), ("bb", 2, 4)])
for o in outputs:
self.assertEqual("babba"[o[1] : o[2]], o[0])
