mirror of
https://github.com/django/django.git
synced 2024-12-25 10:35:48 +00:00
43 lines
1.1 KiB
Python
43 lines
1.1 KiB
Python
|
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
|
||
|
)
|