如何使用assertRaises（）对python类的一个u init_uu（）方法进行单元测试？

Question

我有一节课：

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

并且一个单元测试应该确保传递给构造函数的不正确参数正确地引发一个错误：

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

但是我。。。

NameError: global name 'Error' is not defined

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

Answer 1

这个怎么样：

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