Update test.py

test.py

@@ -382,9 +382,4 @@ async def test_websocket(websocket: WebSocket):
 
     await event_generator(data)
     logging.info("Closing connection")
-    await websocket.close()
-    
-
-
-
-
+    await websocket.close()