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)
