当要测试的函数需要用户输入时，如何制作驱动函数？

def calculateSum(a, b):
    return a + b

def test_calculateSum():
    # 简单驱动函数
    assert calculateSum(3, 5) == 8
    
    # 参数化驱动函数
    test_cases = [
        (1, 2, 3),
        (10, -5, 5),
        (0, 0, 0)
    ]
    
    for a, b, expected in test_cases:
        assert calculateSum(a, b) == expected

# 运行测试
test_calculateSum()

def test_calculateSum_complex():
    test_cases = [
        (1.5, 2.5, 4.0),
        (-3, -7, -10),
        (100, 200, 300)
    ]
    
    for a, b, expected in test_cases:
        assert calculateSum(a, b) == expected

import unittest
from unittest.mock import patch

class TestCalculateSum(unittest.TestCase):
    @patch('builtins.input', side_effect=[3, 5])
    def test_calculateSum_with_input(self, mock_input):
        assert calculateSum(int(input()), int(input())) == 8

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