make State.next() raise keyerror when necessary

2017-06-08 01:09:55 -04:00 · 2017-06-08 01:09:55 -04:00 · 979dc97707
parent e7eeb8d29e
commit 979dc97707
1 changed files with 1 additions and 1 deletions
--- a/stateMachine.py
+++ b/stateMachine.py
@ -54,7 +54,7 @@ class State:

 	def next(self, signal):
 		t = (self, signal)
-		return self if t not in self.stateMachine.transitionTable else self.stateMachine.transitionTable[t]
+		return self.stateMachine.transitionTable[t]
 	
 	def __str__(self):
 		return self.name