要測(cè)試Python函數(shù),您可以使用以下方法:
unittest
模塊創(chuàng)建測(cè)試用例。pytest
。doctest
進(jìn)行測(cè)試。這里有一個(gè)例子,展示了如何使用unittest
模塊測(cè)試一個(gè)簡(jiǎn)單的Python函數(shù):
假設(shè)我們有一個(gè)函數(shù)add_numbers
,它接受兩個(gè)參數(shù)并返回它們的和:
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
模塊測(cè)試這個(gè)函數(shù),請(qǐng)按照以下步驟操作:
unittest
模塊。unittest.TestCase
的測(cè)試類(lèi)。test_
開(kāi)頭。assertEqual
、assertTrue
等方法檢查預(yù)期結(jié)果和實(shí)際結(jié)果是否相符。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()
將上述代碼保存到一個(gè)名為test_add_numbers.py
的文件中,然后在命令行中運(yùn)行該文件。您將看到測(cè)試結(jié)果,顯示所有測(cè)試用例均已通過(guò)。