diff options
Diffstat (limited to 'tests/custom_methods/tests.py')
| -rw-r--r-- | tests/custom_methods/tests.py | 44 |
1 files changed, 44 insertions, 0 deletions
diff --git a/tests/custom_methods/tests.py b/tests/custom_methods/tests.py new file mode 100644 index 0000000000..9d7444ba62 --- /dev/null +++ b/tests/custom_methods/tests.py @@ -0,0 +1,44 @@ +from __future__ import absolute_import + +from datetime import date + +from django.test import TestCase + +from .models import Article + + +class MethodsTests(TestCase): + def test_custom_methods(self): + a = Article.objects.create( + headline="Area man programs in Python", pub_date=date(2005, 7, 27) + ) + b = Article.objects.create( + headline="Beatles reunite", pub_date=date(2005, 7, 27) + ) + + self.assertFalse(a.was_published_today()) + self.assertQuerysetEqual( + a.articles_from_same_day_1(), [ + "Beatles reunite", + ], + lambda a: a.headline, + ) + self.assertQuerysetEqual( + a.articles_from_same_day_2(), [ + "Beatles reunite", + ], + lambda a: a.headline + ) + + self.assertQuerysetEqual( + b.articles_from_same_day_1(), [ + "Area man programs in Python", + ], + lambda a: a.headline, + ) + self.assertQuerysetEqual( + b.articles_from_same_day_2(), [ + "Area man programs in Python", + ], + lambda a: a.headline + ) |