python: Update test bench code for UART.