Fix demo

demo.py

@@ -7,33 +7,42 @@ from leds import LEDs, Color
 frontend = Frontend()
 leds = LEDs()
 
+print("Testing audio output")
 playsound('sample.mp3')
+print("Audio playback ended")
 
 try:
+    print("Testing EEG Frontend")
     data = frontend.read_regs(0x00, 1)
     assert data == [0x3E], "Wrong output"
     print("EEG Frontend responsive")
 
+    print("Testing LEDs")
+    print("Aquisition LED")
     leds.aquisition(True)
     sleep(0.5)
     leds.aquisition(False)
     sleep(0.5)
     leds.aquisition(True)
 
+    print("USER1 (PWM) LED")
     for i in range(200):
         red = (i % 10) * 10
         blue = ((i % 100) // 10) * 10
         leds.led1(red, 0, blue)
         sleep(0.02)
 
-    for state in [RED, BLUE, PURPLE, CLOSED] * 3:
+    print("USER2 (2-color) LED")
+    for state in [Color.RED, Color.BLUE, Color.PURPLE, Color.CLOSED] * 3:
         leds.led2(state)
         sleep(0.2)
 
-    for state in [RED, CLOSED] * 3:
+    print("USER3 LED")
+    for state in [Color.RED, Color.CLOSED] * 3:
         leds.led3(state)
         sleep(0.2)
 
+    print("LEDs testing ended")
 finally:
     frontend.close()
     leds.close()

leds.py

@@ -59,25 +59,25 @@ class LEDs:
         self.led1_B.duty_cycle = blue / 100
 
     def led2(self, value: Color):
-        if value == RED:
+        if value == Color.RED:
             self.led2_R.write(True)
             self.led2_B.write(False)
-        elif value == BLUE:
+        elif value == Color.BLUE:
             self.led2_R.write(False)
             self.led2_B.write(True)
-        elif value == PURPLE:
+        elif value == Color.PURPLE:
             self.led2_R.write(True)
             self.led2_B.write(True)
-        elif value == CLOSED:
+        elif value == Color.CLOSED:
             self.led2_R.write(False)
             self.led2_B.write(False)
         else:
             assert False, "Unknown color"
 
     def led3(self, value: Color):
-        if value == RED:
+        if value == Color.RED:
             self.led3_R.write(True)
-        elif value == CLOSED:
+        elif value == Color.CLOSED:
             self.led3_R.write(False)
         else:
             assert False, "Unknown color"