溫馨提示×

如何在Python中使用單元測試

小樊
93
2024-05-10 13:37:55
欄目: 編程語言

在Python中，可以使用內(nèi)置的unittest模塊來編寫和運(yùn)行單元測試。下面是一個簡單的示例，展示如何在Python中編寫和運(yùn)行單元測試：

import unittest

# 定義一個簡單的函數(shù)
def add(x, y):
    return x + y

# 創(chuàng)建一個測試類，繼承自unittest.TestCase
class TestAddFunction(unittest.TestCase):

    # 編寫測試用例
    def test_add(self):
        self.assertEqual(add(1, 2), 3)
        self.assertEqual(add(-1, 1), 0)
        self.assertEqual(add(0, 0), 0)
        
if __name__ == '__main__':
    unittest.main()

在上面的例子中，我們定義了一個簡單的add函數(shù)，并創(chuàng)建了一個繼承自unittest.TestCase的測試類TestAddFunction。在測試類中，我們編寫了一個名為test_add的測試用例，在這個測試用例中使用assertEqual方法來斷言add函數(shù)的返回值是否符合預(yù)期。

要運(yùn)行這個單元測試，只需將上述代碼保存為一個.py文件，然后在命令行中運(yùn)行該文件即可。運(yùn)行結(jié)果將會輸出每個測試用例的結(jié)果，以及總體的測試結(jié)果。

0 踩