tests/ports/stm32: Tweak tests to run on a wider set of boards.

There should be no change to these tests for existing PYBV1x and PYBD_SFx
boards.

Signed-off-by: Damien George <[email protected]>

Author: dpgeorge

tests/ports/stm32/adc.py

@@ -1,5 +1,10 @@
+import sys
 from pyb import ADC, Timer
 
+if "STM32WB" in sys.implementation._machine:
+    print("SKIP")
+    raise SystemExit
+
 adct = ADC(16)  # Temperature 930 -> 20C
 print(str(adct)[:19])
 adcv = ADC(17)  # Voltage 1500 -> 3.3V

tests/ports/stm32/adcall.py

@@ -1,5 +1,13 @@
+import sys
 from pyb import Pin, ADCAll
 
+if "STM32WB" in sys.implementation._machine:
+    pa0_adc_channel = 5
+    skip_temp_test = True  # temperature fails on WB55
+else:
+    pa0_adc_channel = 0
+    skip_temp_test = False
+
 pins = [Pin.cpu.A0, Pin.cpu.A1, Pin.cpu.A2, Pin.cpu.A3]
 
 # set pins to IN mode, init ADCAll, then check pins are ANALOG
@@ -12,7 +20,7 @@
 # set pins to IN mode, init ADCAll with mask, then check some pins are ANALOG
 for p in pins:
     p.init(p.IN)
-adc = ADCAll(12, 0x70003)
+adc = ADCAll(12, 0x70000 | 3 << pa0_adc_channel)
 for p in pins:
     print(p)
 
@@ -25,7 +33,11 @@
     print(type(adc.read_channel(c)))
 
 # call special reading functions
-print(0 < adc.read_core_temp() < 100)
+print(skip_temp_test or 0 < adc.read_core_temp() < 100)
 print(0 < adc.read_core_vbat() < 4)
 print(0 < adc.read_core_vref() < 2)
 print(0 < adc.read_vref() < 4)
+
+if sys.implementation._build == "NUCLEO_WB55":
+    # Restore button pin settings.
+    Pin("SW", Pin.IN, Pin.PULL_UP)

tests/ports/stm32/extint.py

@@ -1,7 +1,8 @@
 import pyb
 
 # test basic functionality
-ext = pyb.ExtInt("X5", pyb.ExtInt.IRQ_RISING, pyb.Pin.PULL_DOWN, lambda l: print("line:", l))
+pin = pyb.Pin.cpu.A4
+ext = pyb.ExtInt(pin, pyb.ExtInt.IRQ_RISING, pyb.Pin.PULL_DOWN, lambda l: print("line:", l))
 ext.disable()
 ext.enable()
 print(ext.line())

tests/ports/stm32/i2c.py

@@ -1,5 +1,8 @@
-import pyb
-from pyb import I2C
+try:
+    from pyb import I2C
+except ImportError:
+    print("SKIP")
+    raise SystemExit
 
 # test we can correctly create by id
 for bus in (-1, 0, 1):

tests/ports/stm32/i2c_accel.py

@@ -1,15 +1,14 @@
 # use accelerometer to test i2c bus
 
-import pyb
-from pyb import I2C
-
-if not hasattr(pyb, "Accel"):
+try:
+    from pyb import Accel, I2C
+except ImportError:
     print("SKIP")
     raise SystemExit
 
 accel_addr = 76
 
-pyb.Accel()  # this will init the MMA for us
+Accel()  # this will init the MMA for us
 
 i2c = I2C(1, I2C.CONTROLLER, baudrate=400000)
 

tests/ports/stm32/i2c_error.py

@@ -1,12 +1,13 @@
 # test I2C errors, with polling (disabled irqs) and DMA
 
 import pyb
-from pyb import I2C
 
 if not hasattr(pyb, "Accel"):
     print("SKIP")
     raise SystemExit
 
+from pyb import I2C
+
 # init accelerometer
 pyb.Accel()
 

tests/ports/stm32/irq.py

@@ -1,3 +1,4 @@
+import time
 import pyb
 
 
@@ -8,7 +9,7 @@ def test_irq():
     pyb.enable_irq()  # by default should enable IRQ
 
     # check that interrupts are enabled by waiting for ticks
-    pyb.delay(10)
+    time.sleep_ms(10)
 
     # check nested disable/enable
     i1 = pyb.disable_irq()
@@ -18,7 +19,7 @@ def test_irq():
     pyb.enable_irq(i1)
 
     # check that interrupts are enabled by waiting for ticks
-    pyb.delay(10)
+    time.sleep_ms(10)
 
 
 test_irq()

tests/ports/stm32/modstm.py

@@ -1,13 +1,13 @@
 # test stm module
 
 import stm
-import pyb
+import time
 
 # test storing a full 32-bit number
 # turn on then off the A15(=yellow) LED
 BSRR = 0x18
 stm.mem32[stm.GPIOA + BSRR] = 0x00008000
-pyb.delay(100)
+time.sleep_ms(100)
 print(hex(stm.mem32[stm.GPIOA + stm.GPIO_ODR] & 0x00008000))
 stm.mem32[stm.GPIOA + BSRR] = 0x80000000
 print(hex(stm.mem32[stm.GPIOA + stm.GPIO_ODR] & 0x00008000))

tests/ports/stm32/pin.py

@@ -1,14 +1,20 @@
+import sys
 from pyb import Pin
 
-p = Pin("X8", Pin.IN)
+if "PYB" in sys.implementation._machine:
+    test_pin = "X8"
+else:
+    test_pin = Pin.cpu.A7
+
+p = Pin(test_pin, Pin.IN)
 print(p)
 print(p.name())
 print(p.pin())
 print(p.port())
 
-p = Pin("X8", Pin.IN, Pin.PULL_UP)
-p = Pin("X8", Pin.IN, pull=Pin.PULL_UP)
-p = Pin("X8", mode=Pin.IN, pull=Pin.PULL_UP)
+p = Pin(test_pin, Pin.IN, Pin.PULL_UP)
+p = Pin(test_pin, Pin.IN, pull=Pin.PULL_UP)
+p = Pin(test_pin, mode=Pin.IN, pull=Pin.PULL_UP)
 print(p)
 print(p.value())
 

tests/ports/stm32/pyb1.py

@@ -2,6 +2,10 @@
 
 import pyb
 
+if not hasattr(pyb, "delay"):
+    print("SKIP")
+    raise SystemExit
+
 # test delay
 
 pyb.delay(-1)