class TestColorLamp(unittest.TestCase):
def setUp(self):
self.lamp = ColorLamp('white')
def test_set_color(self):
self.lamp.set_color('Red')
self.assertEqual(self.lamp.get_color(), 'Red')
class TestColorLamp(unittest.TestCase):
def setUp(self):
self.lamp = ColorLamp()
def test_default_color(self):
self.assertEqual(self.lamp.get_color(), 'White')
def test_set_color(self):
self.lamp.set_color('Black')
self.assertEqual(self.lamp.get_color(), 'Black')
def setUp(self):
self.lamp = ColorLamp()
def setUp(self):
self.lamp = ColorLamp('white')
def create_ColorLamp(self, id, color):
lamp = ColorLamp(color)
dic = {'ID': id, 'lamp': lamp}
self.lamps.append(dic)
def create_color_lamp(self, lamp_id, color):
self.all_lamps.append(ColorLamp(lamp_id), lamp_id)
self.all_lamps.lamp_list[lamp_id].set_color(color)
def setUp(self):
self.lamp = ColorLamp('Black')
def setUp(self):
self.lamp = LampArray()
self.lamp.lamps.append(ColorLamp())
self.lamp.lamps.append(ColorLamp())