1 files changed, 5 insertions, 5 deletions
diff --git a/tests/template_tests/test_response.py b/tests/template_tests/test_response.py
index b0d66dc809..9fcc0a9c7c 100644
--- a/tests/template_tests/test_response.py
+++ b/tests/template_tests/test_response.py
@@ -77,17 +77,17 @@ class SimpleTemplateResponseTest(SimpleTestCase):
         self.assertFalse(response.is_rendered)
 
         def iteration():
-            for x in response:
-                pass
-        with self.assertRaises(ContentNotRenderedError):
+            list(response)
+
+        msg = 'The response content must be rendered before it can be iterated over.'
+        with self.assertRaisesMessage(ContentNotRenderedError, msg):
             iteration()
         self.assertFalse(response.is_rendered)
 
     def test_iteration_rendered(self):
         # iteration works for rendered responses
         response = self._response().render()
-        res = [x for x in response]
-        self.assertEqual(res, [b'foo'])
+        self.assertEqual(list(response), [b'foo'])
 
     def test_content_access_unrendered(self):
         # unrendered response raises an exception when content is accessed