import pyperf
import socket

s1, s2 = socket.socketpair()
loops = 100
size = 1

def func():
    s1.sendall(b'x' * (size * loops))
    for _ in range(loops):
        s = s2.recv(size)
        if len(s) != size:
            raise Exception

runner = pyperf.Runner()
runner.bench_func('bench', func, inner_loops=loops)

s1.close()
s2.close()

import pyperf
import socket

s1, s2 = socket.socketpair()

def func(size):
    s1.sendall(b'x' * size)
    s = s2.recv(size)
    if len(s) != size:
        raise Exception

runner = pyperf.Runner()
for size in (1, 10, 100, 1000, 10_000, 100_000):
    runner.bench_func(f'recv {size:,}B', func, size)

s1.close()
s2.close()