summaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
authorVytis Banaitis <vytis.banaitis@gmail.com>2017-02-24 03:06:01 +0200
committerTim Graham <timograham@gmail.com>2017-02-23 20:06:01 -0500
commit3dcc3516914f25b58bde9312831f1d94b05cdb53 (patch)
treeac1093e20e9a5b59711000ec42cc4603ca4da664 /tests
parent4cffa9a1ffb37d4de7e99a851a9ed87b3c02d229 (diff)
Refs #23919 -- Used yield from.
Diffstat (limited to 'tests')
-rw-r--r--tests/model_forms/models.py3
-rw-r--r--tests/queries/tests.py3
-rw-r--r--tests/template_tests/filter_tests/test_unordered_list.py8
3 files changed, 4 insertions, 10 deletions
diff --git a/tests/model_forms/models.py b/tests/model_forms/models.py
index 6ed8b2ca06..cf5ca49cf6 100644
--- a/tests/model_forms/models.py
+++ b/tests/model_forms/models.py
@@ -348,8 +348,7 @@ class Colour(models.Model):
name = models.CharField(max_length=50)
def __iter__(self):
- for number in range(5):
- yield number
+ yield from range(5)
def __str__(self):
return self.name
diff --git a/tests/queries/tests.py b/tests/queries/tests.py
index c4181a10b5..86e8405822 100644
--- a/tests/queries/tests.py
+++ b/tests/queries/tests.py
@@ -787,8 +787,7 @@ class Queries1Tests(TestCase):
n_obj = Note.objects.all()[0]
def g():
- for i in [n_obj.pk]:
- yield i
+ yield n_obj.pk
self.assertQuerysetEqual(Note.objects.filter(pk__in=f()), [])
self.assertEqual(list(Note.objects.filter(pk__in=g())), [n_obj])
diff --git a/tests/template_tests/filter_tests/test_unordered_list.py b/tests/template_tests/filter_tests/test_unordered_list.py
index b4d624345b..5e100d5a68 100644
--- a/tests/template_tests/filter_tests/test_unordered_list.py
+++ b/tests/template_tests/filter_tests/test_unordered_list.py
@@ -103,9 +103,7 @@ class FunctionTests(SimpleTestCase):
)
def item_generator():
- yield a
- yield b
- yield c
+ yield from (a, b, c)
self.assertEqual(
unordered_list(item_generator()),
@@ -129,9 +127,7 @@ class FunctionTests(SimpleTestCase):
)
def item_generator():
- yield a
- yield b
- yield c
+ yield from (a, b, c)
self.assertEqual(
unordered_list(item_generator(), autoescape=False),