def setUp(self): self.signal_state = SignalState(1, 1, 2, 3, SignalState.RED)
def testCanStartInAnyState(self): my_signal_state = SignalState(1, 1, 2, 3, SignalState.GREEN) self.assertEqual(SignalState.GREEN, my_signal_state.current_state)