import unittest
def my_function():
raise RuntimeError('hello')
class MyTestCase(unittest.TestCase):
def test_my_function(self):
with self.assertRaises(RuntimeError) as cm:
my_function()
self.assertEqual(str(cm.exception), 'hello')
unittest.main()
aW1wb3J0IHVuaXR0ZXN0CgpkZWYgbXlfZnVuY3Rpb24oKToKICAgIHJhaXNlIFJ1bnRpbWVFcnJvcignaGVsbG8nKQoKY2xhc3MgTXlUZXN0Q2FzZSh1bml0dGVzdC5UZXN0Q2FzZSk6CiAgICBkZWYgdGVzdF9teV9mdW5jdGlvbihzZWxmKToKICAgICAgICB3aXRoIHNlbGYuYXNzZXJ0UmFpc2VzKFJ1bnRpbWVFcnJvcikgYXMgY206CiAgICAgICAgICAgIG15X2Z1bmN0aW9uKCkKICAgICAgICBzZWxmLmFzc2VydEVxdWFsKHN0cihjbS5leGNlcHRpb24pLCAnaGVsbG8nKQoKdW5pdHRlc3QubWFpbigp