Tests: USB: Fix Python 3 compatibility

Use integer division explicitly in basic test.
Convert bytes to str for Python 3 in serial test.
pull/11546/head
Filip Jagodzinski 2019-09-30 10:21:16 +02:00
parent c323af8417
commit 2b4b64fafd
2 changed files with 4 additions and 3 deletions

View File

@ -1376,7 +1376,7 @@ def ep_test_abort(dev, log, verbose=False):
payload_size = (NUM_PACKETS_UNTIL_ABORT + NUM_PACKETS_AFTER_ABORT) * ep_out.wMaxPacketSize payload_size = (NUM_PACKETS_UNTIL_ABORT + NUM_PACKETS_AFTER_ABORT) * ep_out.wMaxPacketSize
num_bytes_written = 0 num_bytes_written = 0
while num_bytes_written < payload_size: while num_bytes_written < payload_size:
payload_out = array.array('B', (num_bytes_written/ep_out.wMaxPacketSize payload_out = array.array('B', (num_bytes_written//ep_out.wMaxPacketSize
for _ in range(ep_out.wMaxPacketSize))) for _ in range(ep_out.wMaxPacketSize)))
try: try:
num_bytes_written += ep_out.write(payload_out) num_bytes_written += ep_out.write(payload_out)

View File

@ -24,6 +24,7 @@ import uuid
import sys import sys
import serial import serial
import serial.tools.list_ports as stlp import serial.tools.list_ports as stlp
import six
import mbed_host_tests import mbed_host_tests
@ -268,7 +269,7 @@ class USBSerialTest(mbed_host_tests.BaseHostTest):
mbed_serial.reset_output_buffer() mbed_serial.reset_output_buffer()
mbed_serial.dtr = True mbed_serial.dtr = True
try: try:
payload = mbed_serial.read(LINE_CODING_STRLEN) payload = six.ensure_str(mbed_serial.read(LINE_CODING_STRLEN))
while len(payload) == LINE_CODING_STRLEN: while len(payload) == LINE_CODING_STRLEN:
baud, bits, parity, stop = (int(i) for i in payload.split(',')) baud, bits, parity, stop = (int(i) for i in payload.split(','))
new_line_coding = { new_line_coding = {
@ -277,7 +278,7 @@ class USBSerialTest(mbed_host_tests.BaseHostTest):
'parity': self._PARITIES[parity], 'parity': self._PARITIES[parity],
'stopbits': self._STOPBITS[stop]} 'stopbits': self._STOPBITS[stop]}
mbed_serial.apply_settings(new_line_coding) mbed_serial.apply_settings(new_line_coding)
payload = mbed_serial.read(LINE_CODING_STRLEN) payload = six.ensure_str(mbed_serial.read(LINE_CODING_STRLEN))
except serial.SerialException as exc: except serial.SerialException as exc:
self.log('TEST ERROR: {}'.format(exc)) self.log('TEST ERROR: {}'.format(exc))
self.notify_complete(False) self.notify_complete(False)