--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/parts/django/tests/modeltests/custom_methods/tests.py Sat Jan 08 11:20:57 2011 +0530
@@ -0,0 +1,42 @@
+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
+ )