def test_waitReset(self):
"""
Tests our wait reset functions
"""
controller = TrafficLightController("TEST")
self.assertEqual(controller.GetRightWait(), 0, msg="Right wait")
controller.IncrementRightWait()
controller.ResetRightWait()
self.assertNotEqual(controller.GetRightWait(), 1, msg="Right wait")
controller.ResetRightWait()
controller.ResetRightWait()
self.assertEqual(controller.GetRightWait(), 0, msg="Right wait")
self.assertEqual(controller.GetOtherWait(), 0, msg="Other wait")
controller.IncrementOtherWait()
controller.ResetOtherWait()
self.assertNotEqual(controller.GetOtherWait(), 1, msg="Other wait")
controller.ResetOtherWait()
controller.ResetOtherWait()
self.assertEqual(controller.GetOtherWait(), 0, msg="Other wait")
def test_wait(self):
"""
Tests our wait increment functions
"""
controller = TrafficLightController("TEST")
self.assertEqual(controller.GetRightWait(), 0)
controller.IncrementRightWait()
controller.IncrementRightWait()
self.assertEqual(controller.GetRightWait(), 2)
controller.ResetRightWait()
self.assertEqual(controller.GetRightWait(), 0)
