Update test_socket.py