要測試Python函數,您可以使用以下方法:
unittest
模塊創建測試用例。pytest
。doctest
進行測試。這里有一個例子,展示了如何使用unittest
模塊測試一個簡單的Python函數:
假設我們有一個函數add_numbers
,它接受兩個參數并返回它們的和:
def add_numbers(a, b):
"""
This function takes two numbers as input and returns their sum.
Parameters:
a (int or float): The first number.
b (int or float): The second number.
Returns:
int or float: The sum of the two numbers.
"""
return a + b
要使用unittest
模塊測試這個函數,請按照以下步驟操作:
unittest
模塊。unittest.TestCase
的測試類。test_
開頭。assertEqual
、assertTrue
等方法檢查預期結果和實際結果是否相符。import unittest
class TestAddNumbers(unittest.TestCase):
def test_add_numbers(self):
self.assertEqual(add_numbers(1, 2), 3)
self.assertEqual(add_numbers(-1, 1), 0)
self.assertEqual(add_numbers(0, 0), 0)
self.assertEqual(add_numbers(3.5, 4.5), 8)
if __name__ == '__main__':
unittest.main()
將上述代碼保存到一個名為test_add_numbers.py
的文件中,然后在命令行中運行該文件。您將看到測試結果,顯示所有測試用例均已通過。