首页 / 知识

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

2023-04-16 21:28:00

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()

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