如何使用assertraisses()对python类的 __init__()方法进行单元测试?

如何使用assertraisses()对python类的 __init__()方法进行单元测试?

How do I unit test an __init__() method of a python class with assertRaises()?

我有一堂课:

1
2
3
4
class MyClass:
def __init__(self, foo):
    if foo != 1:
        raise Error("foo is not equal to 1!")

以及一个单元测试,它应该确保传递给构造函数的参数不正确,从而引发一个错误:

1
2
3
def testInsufficientArgs(self):
    foo = 0
    self.assertRaises((Error), myClass = MyClass(Error, foo))

但我得到…

1
NameError: global name 'Error' is not defined

为什么?我应该在哪里定义这个错误对象?我认为它是作为默认的异常类型内置的,不是吗?


此示例中的"错误"可以是任何异常对象。我想您可能读过一个代码示例,它将其用作元语法占位符,表示"适当的异常类"。

所有异常的基类称为"异常",其大多数子类都是所涉及错误类型的描述性名称,如"oserror"、"valueerror"、"nameerror"、"type error"。

在这种情况下,适当的错误是"value error"(foo的值错误,因此是value error)。我建议将脚本中的"error"替换为"valueerror"。

这里是您试图编写的代码的完整版本,我复制了所有内容,因为您在原始示例中有一个奇怪的关键字参数,您似乎将其与赋值混淆,而我使用的是"failuntil"函数名,因为这是函数的非别名名称:

1
2
3
4
5
6
7
8
9
10
11
12
13
class MyClass:
    def __init__(self, foo):
        if foo != 1:
            raise ValueError("foo is not equal to 1!")

import unittest
class TestFoo(unittest.TestCase):
    def testInsufficientArgs(self):
        foo = 0
        self.failUnlessRaises(ValueError, MyClass, foo)

if __name__ == '__main__':
    unittest.main()

输出是:

1
2
3
4
5
.
----------------------------------------------------------------------
Ran 1 test in 0.007s

OK

单元测试库"UnitTest"中存在其他单元测试框架修复的缺陷。您将注意到,无法从调用上下文访问异常对象。如果要修复此问题,则必须在UnitTest的子类中重新定义该方法:

这是一个正在使用的示例:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
class TestFoo(unittest.TestCase):
    def failUnlessRaises(self, excClass, callableObj, *args, **kwargs):
        try:
            callableObj(*args, **kwargs)
        except excClass, excObj:
            return excObj # Actually return the exception object
        else:
            if hasattr(excClass,'__name__'): excName = excClass.__name__
            else: excName = str(excClass)
            raise self.failureException,"%s not raised" % excName

    def testInsufficientArgs(self):
        foo = 0
        excObj = self.failUnlessRaises(ValueError, MyClass, foo)
        self.failUnlessEqual(excObj[0], 'foo is not equal to 1!')

我从python2.5复制了failUnlessRaises函数unittest.py,并对其进行了轻微修改。


这个怎么样?

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
class MyClass:
    def __init__(self, foo):
        if foo != 1:
            raise Exception("foo is not equal to 1!")

import unittest

class Tests(unittest.TestCase):
    def testSufficientArgs(self):
        foo = 1
        MyClass(foo)

    def testInsufficientArgs(self):
        foo = 2
        self.assertRaises(Exception, MyClass, foo)

if __name__ == '__main__':
    unittest.main()


我认为你在考虑例外。将描述中的"错误"一词替换为"例外",您应该可以继续:—)


推荐阅读