Refs #26022 -- Used context manager version of assertRaises in tests.

1 files changed, 4 insertions, 2 deletions

diff --git a/tests/staticfiles_tests/cases.py b/tests/staticfiles_tests/cases.py
index aedb257351..941c45274a 100644
--- a/tests/staticfiles_tests/cases.py
+++ b/tests/staticfiles_tests/cases.py
@@ -29,7 +29,8 @@ class BaseStaticFilesTestCase(object):
         )
 
     def assertFileNotFound(self, filepath):
-        self.assertRaises(IOError, self._get_file, filepath)
+        with self.assertRaises(IOError):
+            self._get_file(filepath)
 
     def render_template(self, template, **kwargs):
         if isinstance(template, six.string_types):
@@ -46,7 +47,8 @@ class BaseStaticFilesTestCase(object):
         self.assertEqual(self.render_template(template, **kwargs), result)
 
     def assertStaticRaises(self, exc, path, result, asvar=False, **kwargs):
-        self.assertRaises(exc, self.assertStaticRenders, path, result, **kwargs)
+        with self.assertRaises(exc):
+            self.assertStaticRenders(path, result, **kwargs)
 
 
 @override_settings(**TEST_SETTINGS)