Fix unit test failures on Python 3.11 (#869)

1 files changed, 3 insertions, 3 deletions

diff --git a/tests/mobly/snippet/callback_handler_base_test.py b/tests/mobly/snippet/callback_handler_base_test.py
index cda26ef..0891fd5 100644
--- a/tests/mobly/snippet/callback_handler_base_test.py
+++ b/tests/mobly/snippet/callback_handler_base_test.py
@@ -77,16 +77,16 @@ class CallbackHandlerBaseTest(unittest.TestCase):
                                   default_timeout_sec=10)
     self.assertEqual(handler.rpc_max_timeout_sec, 20)
     self.assertEqual(handler.default_timeout_sec, 10)
-    with self.assertRaisesRegex(AttributeError, "can't set attribute"):
+    with self.assertRaises(AttributeError):
       handler.rpc_max_timeout_sec = 5
 
-    with self.assertRaisesRegex(AttributeError, "can't set attribute"):
+    with self.assertRaises(AttributeError):
       handler.default_timeout_sec = 5
 
   def test_callback_id_property(self):
     handler = FakeCallbackHandler(callback_id=MOCK_CALLBACK_ID)
     self.assertEqual(handler.callback_id, MOCK_CALLBACK_ID)
-    with self.assertRaisesRegex(AttributeError, "can't set attribute"):
+    with self.assertRaises(AttributeError):
       handler.callback_id = 'ha'
 
   def test_event_dict_to_snippet_event(self):