diff options
Diffstat (limited to 'tests/test_utils')
| -rw-r--r-- | tests/test_utils/tests.py | 6 |
1 files changed, 4 insertions, 2 deletions
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__) |