创建单元测试两个检查两个psql模式是否相同

import psycopg2

def test_check_schema_equality():
    conn = psycopg2.connect(
        host="your_host",
        port="your_port",
        database="your_database",
        user="your_username",
        password="your_password"
    )
    
    cursor = conn.cursor()
    
    # 查询第一个模式的所有表名
    cursor.execute("SELECT table_name FROM information_schema.tables WHERE table_schema = 'schema1'")
    schema1_tables = cursor.fetchall()
    
    # 查询第二个模式的所有表名
    cursor.execute("SELECT table_name FROM information_schema.tables WHERE table_schema = 'schema2'")
    schema2_tables = cursor.fetchall()
    
    # 检查两个模式的表数量是否相同
    assert len(schema1_tables) == len(schema2_tables), "The number of tables in the two schemas is different."
    
    # 检查两个模式的表名是否相同
    for table1, table2 in zip(schema1_tables, schema2_tables):
        assert table1 == table2, "The table names in the two schemas are different."
    
    conn.close()