import time
from machine import Pin
from time import sleep

red1 = Pin(15, Pin.OUT)
yellow1 = Pin(2, Pin.OUT)
green1 = Pin(4, Pin.OUT)
push = Pin(13, Pin.IN)

while True:
  test = push.value()

  if test == True:
    red1.value(1)
    yellow1.value(0)
    green1.value(0)


  else:
    red1.value(2)
    yellow1.value(0) 
    green1.value(0) 
    time.sleep(1)

    red1.value(0)
    yellow1.value(2) 
    green1.value(0) 
    time.sleep(1) 
 
    red1.value(0) 
    yellow1.value(0) 
    green1.value(2) 
    time.sleep(1)