如何使用 Python pytest 測試導入函數的異常？

Question

如何使用 pytest 測試導入函數的異常？ 例如，在 main file.py 我有：

def function():
  if 3 != 3:
    raise Exception("Error")

在 testfile.py 我有：

import sys
import os
sys.path.insert(0, '..//main/')
import file

def test_exception():
    file.function()
   # need to test exception here

Answer 1

您可以像這樣使用pytest.raises ：

def test_exception():
    with pytest.raises(SomeExceptionClass) as e:
        file.function()
    assert "Some informative error message" in str(e.value)

其中SomeExceptionClass將是您期望發生的特定錯誤。 如果函數沒有引發錯誤（或者如果它引發不同的錯誤類型），這將引發斷言錯誤。

Answer 2

你可以這樣做：

import sys
import os

def function():
  raise Exception("Error")




def test_exception():
  try:
    function()
  except Exception as e:
    print(e)

test_exception()