mirror of https://github.com/ARMmbed/mbed-os.git
40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
from socket import socket, AF_INET, SOCK_DGRAM
|
|
import string, random
|
|
from time import time
|
|
|
|
from private_settings import CLIENT_ADDRESS
|
|
|
|
ECHO_PORT = 7
|
|
|
|
LEN_PACKET = 127
|
|
N_PACKETS = 5000
|
|
TOT_BITS = float(LEN_PACKET * N_PACKETS * 8) * 2
|
|
MEGA = float(1024 * 1024)
|
|
UPDATE_STEP = (N_PACKETS/10)
|
|
|
|
class UDP_EchoClient:
|
|
s = socket(AF_INET, SOCK_DGRAM)
|
|
|
|
def __init__(self, host):
|
|
self.host = host
|
|
self.packet = ''.join(random.choice(string.ascii_uppercase + string.digits) for _ in range(LEN_PACKET))
|
|
|
|
def __packet(self):
|
|
# Comment out the checks when measuring the throughput
|
|
# packet = ''.join(random.choice(string.ascii_uppercase + string.digits) for _ in range(LEN_PACKET))
|
|
UDP_EchoClient.s.sendto(packet, (self.host, ECHO_PORT))
|
|
data = UDP_EchoClient.s.recv(LEN_PACKET)
|
|
# assert packet == data, "packet error:\n%s\n%s\n" % (packet, data)
|
|
|
|
def test(self):
|
|
start = time()
|
|
for i in range(N_PACKETS):
|
|
if (i % UPDATE_STEP) == 0: print '%.2f%%' % ((float(i)/float(N_PACKETS)) * 100.)
|
|
self.__packet()
|
|
t = time() - start
|
|
print 'Throughput: (%.2f)Mbits/s' % ((TOT_BITS / t)/MEGA)
|
|
|
|
while True:
|
|
e = UDP_EchoClient(CLIENT_ADDRESS)
|
|
e.test()
|