def test_displayLightOff(self): """ Test light display function t lights off """ trafficLight = TrafficLight() self.assertEqual(trafficLight.displayLight(False, "T"), " ")
def test_displayLightOnT(self): """ Test light display function t lights on """ trafficLight = TrafficLight() self.assertEqual(trafficLight.displayLight(True, "T"), ">")
def test_displayLightOnS(self): """ Test light display function straight light on """ trafficLight = TrafficLight() self.assertEqual(trafficLight.displayLight(True, "S"), "O")
def test_displayLightOff(self): trafficLight = TrafficLight() self.assertEqual(trafficLight.displayLight(False, "T"), " ")
def test_displayLightOnT(self): trafficLight = TrafficLight() self.assertEqual(trafficLight.displayLight(True, "T"), ">")
def test_displayLightOnS(self): trafficLight = TrafficLight() self.assertEqual(trafficLight.displayLight(True, "S"), "O")