以数据库配置为参数的单元测试Python方法

import unittest
import pymysql

def connect_to_database(host, port, username, password, database):
    # 连接数据库的代码
    connection = pymysql.connect(host=host, port=port, user=username, password=password, db=database)
    return connection

class DatabaseConfigTestCase(unittest.TestCase):
    def test_database_connection(self):
        host = 'localhost'
        port = 3306
        username = 'root'
        password = 'password'
        database = 'mydatabase'

        connection = connect_to_database(host, port, username, password, database)
        self.assertIsNotNone(connection)
        # 其他数据库操作的测试代码

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