def setUp(self):
self.signal_state = SignalState(1, 1, 2, 3, SignalState.RED)
class SignalStateTests(unittest.TestCase):
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)
def testMinTimesSet(self):
self.assertEqual(
self.signal_state.minimum_times.get(SignalState.GREEN), 1)
self.assertEqual(
self.signal_state.minimum_times.get(SignalState.PRIORITY_GREEN), 1)
self.assertEqual(
self.signal_state.minimum_times.get(SignalState.YELLOW), 2)
self.assertEqual(self.signal_state.minimum_times.get(SignalState.RED),
3)
def testOnInitRed(self):
self.assertEqual(self.signal_state.current_state, SignalState.RED)
def testValidStatesChange(self):
# meet the minimum age requirement then change
self.signal_state.changeStateTo(SignalState.RED)
self.signal_state.changeStateTo(SignalState.RED)
self.signal_state.changeStateTo(SignalState.RED)
self.signal_state.changeStateTo(SignalState.RED)
self.signal_state.changeStateTo(SignalState.GREEN)
self.assertEqual(self.signal_state.current_state, SignalState.GREEN)
self.assertEqual(self.signal_state.change_status,
SignalState.STATUS_OK)
def testInvalidStateDoesNotChange(self):
self.signal_state.changeStateTo(SignalState.RED)
self.signal_state.changeStateTo("blue")
self.assertEqual(self.signal_state.current_state, SignalState.RED)
self.assertEqual(self.signal_state.current_state_age, 3)
self.assertEqual(self.signal_state.change_status,
SignalState.STATUS_INVALID_STATE)
def testTransitionMustBeLogical(self):
self.signal_state.changeStateTo(SignalState.RED)
self.signal_state.changeStateTo(SignalState.RED)
self.signal_state.changeStateTo(SignalState.YELLOW)
self.assertEqual(self.signal_state.current_state, SignalState.RED)
self.assertEqual(self.signal_state.change_status,
SignalState.STATUS_INVALID_PROGRESSION)
def testNoStateChangeIfMinTimeNotYetMetForRed(self):
self.signal_state.changeStateTo(SignalState.RED)
self.signal_state.changeStateTo(SignalState.GREEN)
self.assertEqual(self.signal_state.current_state, SignalState.RED)
self.assertEqual(self.signal_state.change_status,
SignalState.STATUS_MIN_TIME)
def testMinTimeNotYetMetForRedIncrementAge(self):
self.signal_state.changeStateTo(SignalState.RED)
self.signal_state.changeStateTo(SignalState.GREEN)
self.assertEqual(self.signal_state.current_state, SignalState.RED)
self.assertEqual(self.signal_state.current_state_age, 3)
def testCurrentStateAgeIsOneOnInit(self):
self.assertEqual(self.signal_state.current_state_age, 1)
def testCurrentStateAgeIsOneIfJustChanged(self):
self.signal_state.changeStateTo(SignalState.RED)
self.signal_state.changeStateTo(SignalState.RED)
self.signal_state.changeStateTo(SignalState.RED)
self.signal_state.changeStateTo(SignalState.GREEN)
self.assertEqual(self.signal_state.current_state_age, 1)
def testCurrentStateAgeIncrementsIfStateStaysTheSame(self):
self.signal_state.changeStateTo(SignalState.RED)
self.assertEqual(self.signal_state.current_state_age, 2)
self.assertEqual(self.signal_state.change_status,
SignalState.STATUS_OK)
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)
class SignalStateTests(unittest.TestCase):
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)
def testMinTimesSet(self):
self.assertEqual(self.signal_state.minimum_times.
get(SignalState.GREEN), 1)
self.assertEqual(
self.signal_state.minimum_times.get(SignalState.PRIORITY_GREEN), 1)
self.assertEqual(self.signal_state.minimum_times.
get(SignalState.YELLOW), 2)
self.assertEqual(self.signal_state.minimum_times.get(SignalState.RED),
3)
def testOnInitRed(self):
self.assertEqual(self.signal_state.current_state, SignalState.RED)
def testValidStatesChange(self):
# meet the minimum age requirement then change
self.signal_state.changeStateTo(SignalState.RED)
self.signal_state.changeStateTo(SignalState.RED)
self.signal_state.changeStateTo(SignalState.RED)
self.signal_state.changeStateTo(SignalState.RED)
self.signal_state.changeStateTo(SignalState.GREEN)
self.assertEqual(self.signal_state.current_state, SignalState.GREEN)
self.assertEqual(self.signal_state.change_status,
SignalState.STATUS_OK)
def testInvalidStateDoesNotChange(self):
self.signal_state.changeStateTo(SignalState.RED)
self.signal_state.changeStateTo("blue")
self.assertEqual(self.signal_state.current_state, SignalState.RED)
self.assertEqual(self.signal_state.current_state_age, 3)
self.assertEqual(self.signal_state.change_status,
SignalState.STATUS_INVALID_STATE)
def testTransitionMustBeLogical(self):
self.signal_state.changeStateTo(SignalState.RED)
self.signal_state.changeStateTo(SignalState.RED)
self.signal_state.changeStateTo(SignalState.YELLOW)
self.assertEqual(self.signal_state.current_state, SignalState.RED)
self.assertEqual(self.signal_state.change_status,
SignalState.STATUS_INVALID_PROGRESSION)
def testNoStateChangeIfMinTimeNotYetMetForRed(self):
self.signal_state.changeStateTo(SignalState.RED)
self.signal_state.changeStateTo(SignalState.GREEN)
self.assertEqual(self.signal_state.current_state, SignalState.RED)
self.assertEqual(self.signal_state.change_status,
SignalState.STATUS_MIN_TIME)
def testMinTimeNotYetMetForRedIncrementAge(self):
self.signal_state.changeStateTo(SignalState.RED)
self.signal_state.changeStateTo(SignalState.GREEN)
self.assertEqual(self.signal_state.current_state, SignalState.RED)
self.assertEqual(self.signal_state.current_state_age, 3)
def testCurrentStateAgeIsOneOnInit(self):
self.assertEqual(self.signal_state.current_state_age, 1)
def testCurrentStateAgeIsOneIfJustChanged(self):
self.signal_state.changeStateTo(SignalState.RED)
self.signal_state.changeStateTo(SignalState.RED)
self.signal_state.changeStateTo(SignalState.RED)
self.signal_state.changeStateTo(SignalState.GREEN)
self.assertEqual(self.signal_state.current_state_age, 1)
def testCurrentStateAgeIncrementsIfStateStaysTheSame(self):
self.signal_state.changeStateTo(SignalState.RED)
self.assertEqual(self.signal_state.current_state_age, 2)
self.assertEqual(self.signal_state.change_status,
SignalState.STATUS_OK)