如何编写python test来连接rabbitmq？

import pika

def test_connect_rabbitmq():
    # 连接RabbitMQ服务器
    connection = pika.BlockingConnection(pika.ConnectionParameters('localhost'))
    channel = connection.channel()

    # 声明一个队列
    channel.queue_declare(queue='test_queue')

    # 发送消息到队列
    channel.basic_publish(exchange='', routing_key='test_queue', body='Hello, RabbitMQ!')

    # 定义一个回调函数来处理接收到的消息
    def callback(ch, method, properties, body):
        print("Received message: %r" % body)

    # 消费队列中的消息
    channel.basic_consume(queue='test_queue', on_message_callback=callback, auto_ack=True)

    # 开始消费
    channel.start_consuming()

    # 关闭连接
    connection.close()