minor fixes

python · kumaraditya303 · Aug 10, 2023 · Aug 10, 2023 · Aug 10, 2023 · 41e269d6e028003e24b03092daa4a28e9d7773c0
commit 41e269d6e028003e24b03092daa4a28e9d7773c0
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()