make State.next() raise keyerror when necessary

This commit is contained in:
petrucci4prez 2017-06-08 01:09:55 -04:00
parent e7eeb8d29e
commit 979dc97707
1 changed files with 1 additions and 1 deletions

View File

@ -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