/tests/modeltests/custom_methods/tests.py

 1from datetime import date
 2
 3from django.test import TestCase
 4
 5from models import Article
 6
 7
 8class MethodsTests(TestCase):
 9    def test_custom_methods(self):
10        a = Article.objects.create(
11            headline="Area man programs in Python", pub_date=date(2005, 7, 27)
12        )
13        b = Article.objects.create(
14            headline="Beatles reunite", pub_date=date(2005, 7, 27)
15        )
16
17        self.assertFalse(a.was_published_today())
18        self.assertQuerysetEqual(
19            a.articles_from_same_day_1(), [
20                "Beatles reunite",
21            ],
22            lambda a: a.headline,
23        )
24        self.assertQuerysetEqual(
25            a.articles_from_same_day_2(), [
26                "Beatles reunite",
27            ],
28            lambda a: a.headline
29        )
30
31        self.assertQuerysetEqual(
32            b.articles_from_same_day_1(), [
33                "Area man programs in Python",
34            ],
35            lambda a: a.headline,
36        )
37        self.assertQuerysetEqual(
38            b.articles_from_same_day_2(), [
39                "Area man programs in Python",
40            ],
41            lambda a: a.headline
42        )