/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

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