Create test_socket.py