async def test_keep_light(self):
handler = MessageHandler()
[x, y, z] = lights.accel_to_color(0, 0, 255)
handler.set_light_data([int(x), int(y), int(z)])
print("Original light:", handler.get_light_data())
handler.set_new_message(True)
result = await lights.keep_light(handler)
self.assertEqual("Finished", result)
self.assertEqual([255, 255, 0], handler.get_light_data())
def test_set_light(self):
handler = MessageHandler()
[x, y, z] = lights.accel_to_color(0, 0, 255)
handler.set_light_data([int(x), int(y), int(z)])
lights.set_light(handler)
self.assertEqual([255, 255, 0], handler.get_light_data())