from codex import *
from time import sleep

delay = 1.0

pressed = buttons.is_pressed(BTN_A)
if pressed:
    pixels.set(0, RED)
    pixels.set(1, RED)
    sleep(delay)
    pixels.set(0, BLACK)
    pixels.set(1, BLACK)
    sleep(delay)

    pixels.set(0, RED)
    pixels.set(1, RED)
    sleep(delay)
    pixels.set(0, BLACK)
    pixels.set(1, BLACK)
    sleep(delay)

    pixels.set(0, RED)
    pixels.set(1, RED)
    sleep(delay)
    pixels.set(0, BLACK)
    pixels.set(1, BLACK)
    sleep(delay)

    pixels.set(0, RED)
    pixels.set(1, RED)
    sleep(delay)
    pixels.set(0, BLACK)
    pixels.set(1, BLACK)
    sleep(delay)

pressed = buttons.is_pressed(BTN_B)
if pressed:
    pixels.set(2, RED)
    pixels.set(3, RED)
    sleep(delay)
    pixels.set(2, BLACK)
    pixels.set(3, BLACK)
    sleep(delay)

    pixels.set(2, RED)
    pixels.set(3, RED)
    sleep(delay)
    pixels.set(2, BLACK)
    pixels.set(3, BLACK)
    sleep(delay)

    pixels.set(2, RED)
    pixels.set(3, RED)
    sleep(delay)
    pixels.set(2, BLACK)
    pixels.set(3, BLACK)
    sleep(delay)

    pixels.set(2, RED)
    pixels.set(3, RED)
    sleep(delay)
    pixels.set(2, BLACK)
    pixels.set(3, BLACK)
    sleep(delay)