minor fixes for asyncio_websockets benchmark (#311)

kumaraditya303 · web-flow · commit ef1d636b7df3 · 2023-08-10T13:26:18.000+05:30
diff --git a/pyperformance/data-files/benchmarks/bm_asyncio_websockets/run_benchmark.py b/pyperformance/data-files/benchmarks/bm_asyncio_websockets/run_benchmark.py
@@ -1,23 +1,24 @@
 """
 Benchmark for asyncio websocket server and client performance
-transferring 10MB of data.
+transferring 1MB of data.
+
+Author: Kumar Aditya
 """
 
 import pyperf
 import websockets.server
 import websockets.client
 import asyncio
 
-CHUNK_SIZE = 1024
+CHUNK_SIZE = 1024 ** 2
 DATA = b"x" * CHUNK_SIZE
 
 stop: asyncio.Event
 
 
 async def handler(websocket) -> None:
     for _ in range(100):
-        data = await websocket.recv()
-        assert data == DATA
+        await websocket.recv()
 
     stop.set()
 
