fix: remove get_state method

This commit is contained in:
2026-07-16 06:34:56 +03:00
parent 677c02b900
commit ac201c0854
2 changed files with 5 additions and 20 deletions
+5 -15
View File
@@ -41,16 +41,6 @@ class TestMachine(unittest.TestCase):
with self.assertRaises(ValueError):
m.add_state("A", False, True)
# GET STATE
def test_get_states(self):
m = Machine()
m.add_state("A", True, False)
state = m.get_state("A")
self.assertIsNotNone(state)
assert state is not None
self.assertEqual(state.get_name(), "A")
# ADD TRANSITION
def test_add_transition(self):
@@ -200,19 +190,19 @@ class TestMachine(unittest.TestCase):
def test_reset(self):
m = Machine()
m.add_state("A", True, False)
m.add_state("B", False, True)
m.add_transition("X", "A", "B", lambda context: print("A -> B"))
m.add_state("C", True, False)
m.add_state("D", False, True)
m.add_transition("X", "C", "D", lambda context: print("A -> B"))
m.process_input("X")
current_state = m.get_current_state()
self.assertIsNotNone(current_state)
assert current_state is not None
self.assertEqual(current_state.get_name(), "B")
self.assertEqual(current_state.get_name(), "D")
m.reset()
current_state = m.get_current_state()
self.assertIsNotNone(current_state)
assert current_state is not None
self.assertEqual(current_state.get_name(), "A")
self.assertEqual(current_state.get_name(), "C")
def test_reset_no_start_state(self):
m = Machine()