from machine import Pin, I2C
import sh1106

# ESP32 Pin assignment 
i2c = I2C(0, scl=Pin(22), sda=Pin(21))
x = 0
y = 0
u = 1
m = 1
oled_width = 128
oled_height = 64
oled = sh1106.SH1106_I2C(oled_width, oled_height, i2c)
oled.flip()
while True:
    oled.fill(0)
    oled.rect(x,y, 10, 10, 1)      
    oled.show()
    x = x + u
    y = y + m
   
    if  x>116 or x<0:
        u = -u
       
    if y>54 or y<0:
        m = -m