python 读取数据表数据（Dtt+Pytest）

import pymysql
import pytest


def get_mysql_data():
    # 打开数据库连接
    db = pymysql.connect(host='127.0.0.1', database='test_db', port=3306, user='root', password='123456')
    # 获取数据库游标
    cursor = db.cursor()
    # 执行sql语句
    sql = "select name,age,six from user_info"
    cursor.execute(sql)
    # 返回所有查询结果
    lst_data = cursor.fetchall()
　　cursor.close() # 关闭游标
　　db.close() # 关闭数据库连接
    return lst_data


@pytest.mark.parametrize('name,age,six', get_mysql_data())
def test_mysql(name, age, six):
    print(name, age, six)


if __name__ == '__main__':
    # print(get_mysql_data())
    pytest.main(['-vs', 'test_mysql.py'])