python unittest.TestCase.assertRaises 不起作用

我正在尝试在 Python 中对“add”函数运行测试，但出现错误：

7
E
======================================================================
ERROR: test_upper (__main__.TestStringMethods)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "C:/Users/MaZee/PycharmProjects/andela/asdasd.py", line 22, in test_upper
    self.assertEqual("Input should be a string:", cm.exception.message , "Input is not a string:")
AttributeError: '_AssertRaisesContext' object has no attribute 'exception'

----------------------------------------------------------------------
Ran 1 test in 0.001s

FAILED (errors=1)

Process finished with exit code 1

这是我的代码：

    import unittest

    def add(a,b):
        """
        Returns the addition value of a and b.
        """
        try:
          out = a + b
        except TypeError:
          raise TypeError("Input should be a string:")

        print (out)
        return



class TestStringMethods(unittest.TestCase):

    def test_upper(self):
        with self.assertRaises(TypeError) as cm:
           add(3,4)
           self.assertEqual("Input should be a string:", cm.exception.message , "Input is not a string:")


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

正如错误消息告诉您的那样，您的断言加注对象没有属性exception。更具体地说，这个调用：

cm.exception.message

cm在这种情况下是你的断言对象，并且因为你正在测试的代码实际上从未引发，所以你的cm对象不会有exception您尝试访问的属性。

现在，我们来看看为什么会发生这种情况。您正在尝试测试当exception正在你的身边长大add方法，为了提高TypeError。但是，如果您查看测试用例，则会将两个有效整数传递给add方法。您不会引发异常，因为这是一个有效的测试用例。

对于您的单元测试，您希望测试当您raise一些东西，即插入无效数据到add方法。再次尝试您的代码，但这一次在单元测试中，传递以下内容：

add(5, 'this will raise')

您现在将得到您的TypeError.

您还需要在上下文管理器之外执行断言验证：

def test_upper(self):
    with self.assertRaises(TypeError) as cm:
        add(3, 'this will raise')
    self.assertEqual("Input should be a string:", cm.exception.message, "Input is not a string:")

现在你会遇到另一个问题。没有message属性。你应该简单地检查一下cm.exception。此外，在你的add你的字符串的方法是：

"Input should be a string:"

但是，您正在检查它是否是：

"Input is not a string:"

所以，一旦你更正你的单元测试以使用cm.exception，您现在将面临：

AssertionError: 'Input should be a string:' != TypeError('Input should be a string:',) : Input is not a string:

因此，您的断言应该通过调用来检查异常字符串str on the cm.exception:

self.assertEqual("Input should be a string:", str(cm.exception), "Input should be a string:")

所以，你的完整测试方法应该是：

def test_upper(self):
    with self.assertRaises(TypeError) as cm:
        add(3, 'this will raise')
    self.assertEqual("Input should be a string:", str(cm.exception), "Input should be a string:")