input: Write a python function to create a TCP client that connects to a server, sends a message, and receives a response.
  assert tcp_client('127.0.0.1', 65432, 'Hello, world') == b'Hello, world'
  assert tcp_client('127.0.0.1', 65432, 'Test message') == b'Test response'
  assert tcp_client('127.0.0.1', 65432, 'Ping') == b'Pong'
output: import socket
  
  def tcp_client(host, port, message):
      with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
          s.connect((host, port))
          s.sendall(message.encode())
          response = s.recv(1024)
      return response