def test_output_stuff(self):
f = MutableFST()
s0 = f.get_initial_state()
self.assertEqual(f.get_outputs(s0), set())
self.assertEqual(f.to_dict(), {"initial": s0, "finals": [], "transitions": {s0: {}}})
f.add_output(s0, 1)
self.assertEqual(f.get_outputs(s0), {1})
self.assertEqual(f.to_dict(), {"initial": s0, "finals": [], "outputs": {s0: [1]}, "transitions": {s0: {}}})
f.add_output(s0, 2)
self.assertEqual(f.get_outputs(s0), {1, 2})
self.assertEqual(f.to_dict(), {"initial": s0, "finals": [], "outputs": {s0: [1, 2]}, "transitions": {s0: {}}})
f.add_output(s0, 1)
self.assertEqual(f.get_outputs(s0), {1, 2})
self.assertEqual(f.to_dict(), {"initial": s0, "finals": [], "outputs": {s0: [1, 2]}, "transitions": {s0: {}}})
