python测试mysql数据库

import pymysql

try:
    conn = pymysql.connect(host='localhost', user='root', password='password', db='testdb')
except pymysql.Error as e:
    print(f"连接失败: {e}")

try:
    with conn.cursor() as cursor:
        sql = "SELECT * FROM users WHERE id = %s"
        cursor.execute(sql, (1,))
        result = cursor.fetchall()
except pymysql.Error as e:
    print(f"SQL执行错误: {e}")

import unittest
from pymysql import connect

class TestDatabase(unittest.TestCase):
    def setUp(self):
        self.conn = connect(host='localhost', user='root', password='password', db='testdb')
        self.cursor = self.conn.cursor()

    def test_select(self):
        sql = "SELECT * FROM users WHERE id = %s"
        self.cursor.execute(sql, (1,))
        result = self.cursor.fetchall()
        self.assertEqual(len(result), 1)

    def tearDown(self):
        self.cursor.close()
        self.conn.close()

if __name__ == '__main__':
    unittest.main()