100% test coverage

2026-07-15 17:13:07 +03:00
parent 159ad3b140
commit 000495e055
2 changed files with 16 additions and 10 deletions
@@ -233,4 +233,16 @@ class TestMachine(unittest.TestCase):
        t = Transition("TEST_1", "A", "B", lambda context: print("A -> B"))
        self.assertEqual(str(t), "[Transition (A, B, TEST_1)]")

+    # MACHINE
+
+    def test_machine_str(self):
+        m = Machine()
+        m.add_state("A", True, False)
+        m.add_state("B", False, False)
+        m.add_state("C", False, True)
+        m.add_transition("TEST_1", "A", "B", lambda context: print("A -> B"))
+        m.add_transition("TEST_2", "B", "C", lambda context: print("B -> C"))
+        result = "=MACHINE=\n\tSTATES\n\t\t[State A]\n\t\t[State B]\n\t\t[State C]\n\tTRANSITIONS\n\t\t[Transition (A, B, TEST_1)]\n\t\t[Transition (B, C, TEST_2)]"
+        self.assertEqual(str(m), result)
+