/tests/modeltests/custom_methods/tests.py
https://code.google.com/p/mango-py/ · Python · 42 lines · 36 code · 6 blank · 0 comment · 0 complexity · 66412bab3ecdbdcd818d2a80039cff5f MD5 · raw file
- 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
- )