From 5627057f63f30bda6bbef7e8169c2a6156542f90 Mon Sep 17 00:00:00 2001 From: Jon Dufresne Date: Sun, 28 Apr 2019 06:15:19 -0700 Subject: Refs #26022 -- Used context manager version of assertRaisesMessage in tests. Follow up to 253adc2b8a52982139d40c4f55b3fd446e1cb8f3. --- tests/test_utils/tests.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) (limited to 'tests/test_utils') diff --git a/tests/test_utils/tests.py b/tests/test_utils/tests.py index e953eb6609..69a99d47d2 100644 --- a/tests/test_utils/tests.py +++ b/tests/test_utils/tests.py @@ -32,9 +32,11 @@ class SkippingTestCase(SimpleTestCase): def _assert_skipping(self, func, expected_exc, msg=None): try: if msg is not None: - self.assertRaisesMessage(expected_exc, msg, func) + with self.assertRaisesMessage(expected_exc, msg): + func() else: - self.assertRaises(expected_exc, func) + with self.assertRaises(expected_exc): + func() except unittest.SkipTest: self.fail('%s should not result in a skipped test.' % func.__name__) -- cgit v1.3