kopia lustrzana https://github.com/felHR85/UsbSerial
35 wiersze
1.0 KiB
Python
35 wiersze
1.0 KiB
Python
|
# UsbSerial test: Integration test
|
||
|
# args:
|
||
|
# port (eg /dev/ttyUSB0)
|
||
|
# speed in bauds (eg 115200)
|
||
|
|
||
|
import serial
|
||
|
import sys
|
||
|
import os
|
||
|
|
||
|
class style():
|
||
|
RED = lambda x: '\033[31m' + str(x)
|
||
|
GREEN = lambda x: '\033[32m' + str(x)
|
||
|
BLUE = lambda x: '\033[34m' + str(x)
|
||
|
RESET = lambda x: '\033[0m' + str(x)
|
||
|
|
||
|
port = sys.argv[1]
|
||
|
speed = sys.argv[2]
|
||
|
|
||
|
test_sizes = [1024, 2048, 16384]
|
||
|
|
||
|
for i in range(0,3):
|
||
|
comm = serial.Serial(port, int(speed))
|
||
|
print("Creating " + str(test_sizes[i]) + " bytes buffer")
|
||
|
data_tx = os.urandom(test_sizes[i])
|
||
|
print("Sending buffer of " + str(test_sizes[i]) + " bytes")
|
||
|
comm.write(data_tx)
|
||
|
print("Receiving " + str(test_sizes[i]) + " bytes buffer")
|
||
|
data_rx = comm.read(test_sizes[i])
|
||
|
|
||
|
if data_tx == data_rx:
|
||
|
print(style.GREEN("Success: " + str(test_sizes[i]) + " bytes buffer was transmitted correctly"))
|
||
|
else:
|
||
|
print(style.RED("Error: " + str(test_sizes[i]) + " bytes buffer was not transmitted correctly"))
|
||
|
print(style.RESET(""))
|