python - 测试是否引发了 ValidationError

Question

我想测试是否引发了异常，我该怎么做？

在我的 models.py 中我有这个函数，我想测试的那个:

  def validate_percent(value):
    if not (value >= 0 and value <= 100):
      raise ValidationError('error')

在我的 tests.py 中我试过这个:

def test_validate_percent(self):
    self.assertRaises(ValidationError, validate_percent(1000))

测试的输出是:

..E
======================================================================
ERROR: test_validate_percent (tm.tests.models.helpers.HelpersTestCase)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "/...py", line 21, in test_validate_percent
    self.assertRaises(ValidationError, validate_percent(1000))
  File "/....py", line 25, in validate_percent
    raise ValidationError(u'error' % value)
ValidationError: ['error']

Answer 1

assertRaises 用作上下文管理器:

def test_validate_percent(self):
    with self.assertRaises(ValidationError):
        validate_percent(1000)

或者调用一个:

def test_validate_percent(self):
    self.assertRaises(ValidationError, validate_percent, 1000)

• http://docs.python.org/2/library/unittest.html#unittest.TestCase.assertRaises