circuitpython/tests/wipy/i2c.py

176 lines
4.1 KiB
Python
Raw Normal View History

"""
2015-09-09 16:42:42 -04:00
I2C test for the CC3200 based boards.
A MPU-9150 sensor must be connected to the I2C bus.
"""
2015-09-09 16:42:42 -04:00
2015-09-27 07:45:48 -04:00
from machine import I2C
2015-09-09 16:42:42 -04:00
import os
import time
2015-09-09 16:42:42 -04:00
2015-09-27 07:45:48 -04:00
mch = os.uname().machine
if "LaunchPad" in mch:
i2c_pins = ("GP11", "GP10")
elif "WiPy" in mch:
i2c_pins = ("GP15", "GP10")
2015-09-09 16:42:42 -04:00
else:
raise Exception("Board not supported!")
2015-09-09 16:42:42 -04:00
i2c = I2C(0, I2C.MASTER, baudrate=400000)
# try initing without the peripheral id
i2c = I2C()
print(i2c)
i2c = I2C(mode=I2C.MASTER, baudrate=50000, pins=i2c_pins)
print(i2c)
2015-09-09 16:42:42 -04:00
i2c = I2C(0, I2C.MASTER, baudrate=100000)
print(i2c)
i2c = I2C(0, mode=I2C.MASTER, baudrate=400000)
print(i2c)
i2c = I2C(0, mode=I2C.MASTER, baudrate=400000, pins=i2c_pins)
print(i2c)
addr = i2c.scan()[0]
print(addr)
reg = bytearray(1)
reg2 = bytearray(2)
reg2_r = bytearray(2)
# reset the sensor
reg[0] |= 0x80
print(1 == i2c.writeto_mem(addr, 107, reg))
time.sleep_ms(100) # wait for the sensor to reset...
2015-09-09 16:42:42 -04:00
print(1 == i2c.readfrom_mem_into(addr, 107, reg)) # read the power management register 1
2015-09-09 16:42:42 -04:00
print(0x40 == reg[0])
# now just read one byte
data = i2c.readfrom_mem(addr, 117, 1) # read the "who am I?" register
2015-09-09 16:42:42 -04:00
print(0x68 == data[0])
print(len(data) == 1)
print(1 == i2c.readfrom_mem_into(addr, 117, reg)) # read the "who am I?" register again
2015-09-09 16:42:42 -04:00
print(0x68 == reg[0])
# now try reading two bytes
data = i2c.readfrom_mem(addr, 116, 2) # read the "who am I?" register
2015-09-09 16:42:42 -04:00
print(0x68 == data[1])
print(data == b"\x00\x68")
2015-09-09 16:42:42 -04:00
print(len(data) == 2)
print(2 == i2c.readfrom_mem_into(addr, 116, reg2)) # read the "who am I?" register again
2015-09-09 16:42:42 -04:00
print(0x68 == reg2[1])
print(reg2 == b"\x00\x68")
2015-09-09 16:42:42 -04:00
print(1 == i2c.readfrom_mem_into(addr, 107, reg)) # read the power management register 1
2015-09-09 16:42:42 -04:00
print(0x40 == reg[0])
# clear the sleep bit
reg[0] = 0
print(1 == i2c.writeto_mem(addr, 107, reg))
# read it back
i2c.readfrom_mem_into(addr, 107, reg)
print(0 == reg[0])
# set the sleep bit
reg[0] = 0x40
print(1 == i2c.writeto_mem(addr, 107, reg))
# read it back
i2c.readfrom_mem_into(addr, 107, reg)
print(0x40 == reg[0])
# reset the sensor
reg[0] |= 0x80
print(1 == i2c.writeto_mem(addr, 107, reg))
time.sleep_ms(100) # wait for the sensor to reset...
2015-09-09 16:42:42 -04:00
# now read and write two register at a time
print(2 == i2c.readfrom_mem_into(addr, 107, reg2))
print(0x40 == reg2[0])
print(0x00 == reg2[1])
# clear the sleep bit
reg2[0] = 0
# set some other bits
reg2[1] |= 0x03
print(2 == i2c.writeto_mem(addr, 107, reg2))
# read it back
i2c.readfrom_mem_into(addr, 107, reg2_r)
print(reg2 == reg2_r)
# reset the sensor
reg[0] = 0x80
print(1 == i2c.writeto_mem(addr, 107, reg))
time.sleep_ms(100) # wait for the sensor to reset...
2015-09-09 16:42:42 -04:00
# try some raw read and writes
reg[0] = 117 # register address
print(1 == i2c.writeto(addr, reg, stop=False)) # just write the register address
2015-09-09 16:42:42 -04:00
# now read
print(1 == i2c.readfrom_into(addr, reg))
print(reg[0] == 0x68)
reg[0] = 117 # register address
print(1 == i2c.writeto(addr, reg, stop=False)) # just write the register address
2015-09-09 16:42:42 -04:00
# now read
print(0x68 == i2c.readfrom(addr, 1)[0])
i2c.readfrom_mem_into(addr, 107, reg2)
print(0x40 == reg2[0])
print(0x00 == reg2[1])
reg2[0] = 107 # register address
2015-09-09 16:42:42 -04:00
reg2[1] = 0
print(2 == i2c.writeto(addr, reg2, stop=True)) # write the register address and the data
i2c.readfrom_mem_into(addr, 107, reg) # check it back
2015-09-09 16:42:42 -04:00
print(reg[0] == 0)
# check for memory leaks...
for i in range(0, 1000):
2015-09-09 16:42:42 -04:00
i2c = I2C(0, I2C.MASTER, baudrate=100000)
# test deinit
i2c = I2C(0, I2C.MASTER, baudrate=100000)
i2c.deinit()
print(i2c)
# next ones should raise
try:
i2c.scan()
except Exception:
print("Exception")
try:
i2c.readfrom(addr, 1)
except Exception:
print("Exception")
try:
i2c.readfrom_into(addr, reg)
except Exception:
print("Exception")
try:
i2c.readfrom_mem_into(addr, 107, reg)
except Exception:
print("Exception")
try:
i2c.writeto(addr, reg, stop=False)
except Exception:
print("Exception")
try:
i2c.writeto_mem(addr, 107, reg)
except Exception:
print("Exception")
try:
i2c.readfrom_mem(addr, 116, 2)
except Exception:
print("Exception")
2015-09-14 13:38:28 -04:00
try:
I2C(1, I2C.MASTER, baudrate=100000)
except Exception:
print("Exception")
2015-09-09 16:42:42 -04:00
# reinitialization must work
i2c.init(baudrate=400000)
print(i2c)