diff options
Diffstat (limited to 'tests/custom_methods/tests.py')
| -rw-r--r-- | tests/custom_methods/tests.py | 16 |
1 files changed, 10 insertions, 6 deletions
diff --git a/tests/custom_methods/tests.py b/tests/custom_methods/tests.py index bbcec1f9f2..2bae2d2623 100644 --- a/tests/custom_methods/tests.py +++ b/tests/custom_methods/tests.py @@ -16,27 +16,31 @@ class MethodsTests(TestCase): self.assertFalse(a.was_published_today()) self.assertQuerysetEqual( - a.articles_from_same_day_1(), [ + a.articles_from_same_day_1(), + [ "Beatles reunite", ], lambda a: a.headline, ) self.assertQuerysetEqual( - a.articles_from_same_day_2(), [ + a.articles_from_same_day_2(), + [ "Beatles reunite", ], - lambda a: a.headline + lambda a: a.headline, ) self.assertQuerysetEqual( - b.articles_from_same_day_1(), [ + b.articles_from_same_day_1(), + [ "Parrot programs in Python", ], lambda a: a.headline, ) self.assertQuerysetEqual( - b.articles_from_same_day_2(), [ + b.articles_from_same_day_2(), + [ "Parrot programs in Python", ], - lambda a: a.headline + lambda a: a.headline, ) |