From 3f54e5dff27a135dff8a53a0483e5bb3a8b9a522 Mon Sep 17 00:00:00 2001
From: Damien George <damien@micropython.org>
Date: Mon, 21 Oct 2024 11:19:03 +1100
Subject: [PATCH] tests/extmod: Support esp32,mimxrt,stm32,samd ports in UART
 TX test.

Getting this test running on stm32- and mimxrt-based boards requires adding
a small delay after constructing the UART so that the initial idle frame
has time to be transmitted before the test starts.

Also, the timing margin needs to account for an additional 1-bit worth of
time on some MCUs.

Thanks to @robert-hh for the esp32, mimxrt and samd settings.

Signed-off-by: Damien George <damien@micropython.org>
---
 tests/extmod/machine_uart_tx.py | 34 ++++++++++++++++++++++++++++-----
 1 file changed, 29 insertions(+), 5 deletions(-)

diff --git a/tests/extmod/machine_uart_tx.py b/tests/extmod/machine_uart_tx.py
index 25bb834929..e9652f3c7a 100644
--- a/tests/extmod/machine_uart_tx.py
+++ b/tests/extmod/machine_uart_tx.py
@@ -9,12 +9,34 @@ except ImportError:
 
 import time, sys
 
+initial_delay_ms = 0
+bit_margin = 0
+timing_margin_us = 100
+
 # Configure pins based on the target.
-if "rp2" in sys.platform:
+if "esp32" in sys.platform:
+    uart_id = 1
+    pins = {}
+    timing_margin_us = 400
+elif "mimxrt" in sys.platform:
+    uart_id = 1
+    pins = {}
+    initial_delay_ms = 20  # UART sends idle frame after init, so wait for that
+    bit_margin = 1
+elif "pyboard" in sys.platform:
+    uart_id = 4
+    pins = {}
+    initial_delay_ms = 50  # UART sends idle frame after init, so wait for that
+    bit_margin = 1  # first start-bit must wait to sync with the UART clock
+elif "rp2" in sys.platform:
     uart_id = 0
-    tx_pin = "GPIO0"
-    rx_pin = "GPIO1"
+    pins = {"tx": "GPIO0", "rx": "GPIO1"}
     timing_margin_us = 180
+elif "samd" in sys.platform:
+    uart_id = 2
+    pins = {"tx": "D1", "rx": "D0"}
+    timing_margin_us = 300
+    bit_margin = 1
 else:
     print("SKIP")
     raise SystemExit
@@ -22,7 +44,8 @@ else:
 # Test that write+flush takes the expected amount of time to execute.
 for bits_per_s in (2400, 9600, 115200):
     text = "Hello World"
-    uart = UART(uart_id, bits_per_s, bits=8, parity=None, stop=1, tx=tx_pin, rx=rx_pin)
+    uart = UART(uart_id, bits_per_s, bits=8, parity=None, stop=1, **pins)
+    time.sleep_ms(initial_delay_ms)
 
     start_us = time.ticks_us()
     uart.write(text)
@@ -33,4 +56,5 @@ for bits_per_s in (2400, 9600, 115200):
     bits_per_char = 10
     expect_us = (len(text)) * bits_per_char * 1_000_000 // bits_per_s
     delta_us = abs(duration_us - expect_us)
-    print(bits_per_s, delta_us <= timing_margin_us or delta_us)
+    margin_us = timing_margin_us + bit_margin * 1_000_000 // bits_per_s
+    print(bits_per_s, delta_us <= margin_us or delta_us)