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: {}}})
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: {}}})