是的，Python的assert語句可以用于單元測試。在Python中，assert語句用于檢查一個條件是否為真。如果條件為假，則會引發一個AssertionError異常。在編寫單元測試時，我們通常使用assert語句來驗證代碼的輸出是否符合預期。
以下是一個簡單的示例，展示了如何在單元測試中使用assert語句：
import unittest
def add(a, b):
return a + b
class TestAddition(unittest.TestCase):
def test_add(self):
self.assertEqual(add(1, 2), 3)
self.assertEqual(add(-1, 1), 0)
self.assertEqual(add(0, 0), 0)
# 使用assert語句檢查異常情況
with self.assertRaises(TypeError):
add("1", 2)
if __name__ == "__main__":
unittest.main()
在這個例子中，我們定義了一個add函數，用于計算兩個數的和。然后，我們創建了一個名為TestAddition的單元測試類，該類繼承自unittest.TestCase。在test_add方法中，我們使用assertEqual方法來驗證add函數的輸出是否符合預期。此外，我們還使用了assertRaises方法來檢查當傳遞錯誤類型的參數時，add函數是否會引發TypeError異常。
