python unittest 中断言 self.assertRaisesRegex 的使用中关于 erro message 正则使用的踩坑记录

最新推荐文章于 2024-07-31 21:53:35 发布

qq_16069927

最新推荐文章于 2024-07-31 21:53:35 发布

阅读量987

点赞数

文章标签： unittest python

本文链接：https://blog.csdn.net/qq_16069927/article/details/113243128

版权

该博客源自于 https://stackoverflow.com/questions/58723899/how-to-properly-use-self-assertraisesregex-as-a-context-manager

import unittest


def fn():
    raise KeyError('my message')


class Test(unittest.TestCase):
    def test_passes(self):
        with self.assertRaisesRegex(KeyError, 'my message'):
            fn()

        with self.assertRaisesRegex(KeyError, 'my'):
            fn()

    def test_fails(self):
        # Why does this test fail?
        # Isn't '^my message$' a valid regex?
        # How do I check if the message is exactly 'my message'?
        with self.assertRaisesRegex(KeyError, '^my message$'):
            fn()


unittest.main()

上述代码的测试在执行 test_fails 时出现报错：

AssertionError: "^my message$" does not match "'my message'"

因为在正在匹配的时候，会包含引号，具体的可以通过以下方式来查看具体的 message

>>> str(KeyError('my message'))
"'my message'"

处理方式：代码改为以下方式即可使用

import unittest


def fn():
    raise KeyError('my message')


class Test(unittest.TestCase):
    def test_passes(self):
        with self.assertRaisesRegex(KeyError, 'my message'):
            fn()

        with self.assertRaisesRegex(KeyError, 'my'):
            fn()

    def test_fails(self):
        # Why does this test fail?
        # Isn't '^my message$' a valid regex?
        # How do I check if the message is exactly 'my message'?
        with self.assertRaisesRegex(KeyError, '^\'my message\'$'):
            fn()


unittest.main()