def test_add_transition(self): f = MutableFST() s0 = f.get_initial_state() s1 = f.add_state() s2 = f.add_state() # s0 a s1 f.add_transition(s0, "a", s1) self.assertEqual(f.to_dict(), {"initial": s0, "finals": [], "transitions": {s0: {"a": s1}, s1: {}, s2: {}}}) t = f.get_target(s0, "a") self.assertEqual(t, s1) self.assertEqual(f.get_letters(), ["a"]) # s0 b s2 f.add_transition(s0, "b", s2) self.assertEqual( f.to_dict(), {"initial": s0, "finals": [], "transitions": {s0: {"a": s1, "b": s2}, s1: {}, s2: {}}} ) t = f.get_target(s0, "b") self.assertEqual(t, s2) self.assertEqual(f.get_letters(), ["a", "b"]) # s1 a s0 f.add_transition(s1, "a", s0) self.assertEqual( f.to_dict(), {"initial": s0, "finals": [], "transitions": {s0: {"a": s1, "b": s2}, s1: {"a": s0}, s2: {}}} ) t = f.get_target(s1, "a") self.assertEqual(t, s0) # s0 a s2 f.add_transition(s0, "a", s2) self.assertEqual( f.to_dict(), {"initial": s0, "finals": [], "transitions": {s0: {"a": s2, "b": s2}, s1: {"a": s0}, s2: {}}} ) t = f.get_target(s0, "a") self.assertEqual(t, s2)
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])