Example #1
0
 def test_add_state(self):
     f = MutableFST()
     s0 = f.get_initial_state()
     s1 = f.add_state()
     self.assertEqual(f.get_states(), [s0, s1])
     self.assertEqual(f.to_dict(), {"initial": s0, "finals": [], "transitions": {s0: {}, s1: {}}})
     s2 = f.add_state()
     self.assertEqual(f.get_states(), [s0, s1, s2])
     self.assertEqual(f.to_dict(), {"initial": s0, "finals": [], "transitions": {s0: {}, s2: {}, s1: {}}})
Example #2
0
 def test_construction(self):
     f = MutableFST()
     self.assertEqual(f.get_letters(), [])
     s0 = f.get_initial_state()
     self.assertEqual(s0, 0)
     self.assertEqual(f.get_states(), [s0])
     self.assertEqual(f.get_final_states(), [])
     self.assertEqual(f.to_dict(), {"initial": s0, "finals": [], "transitions": {s0: {}}})