From 1933e56eca1ad17de7dd133bfb7cbee9858a75a3 Mon Sep 17 00:00:00 2001 From: Sergey Fedoseev Date: Sat, 9 Feb 2019 19:18:48 +0500 Subject: Removed uneeded generator expressions and list comprehensions. --- tests/template_tests/test_response.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) (limited to 'tests/template_tests') 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 -- cgit v1.3