summaryrefslogtreecommitdiff
path: root/tests/datetimes
diff options
context:
space:
mode:
authordjango-bot <ops@djangoproject.com>2022-02-03 20:24:19 +0100
committerMariusz Felisiak <felisiak.mariusz@gmail.com>2022-02-07 20:37:05 +0100
commit9c19aff7c7561e3a82978a272ecdaad40dda5c00 (patch)
treef0506b668a013d0063e5fba3dbf4863b466713ba /tests/datetimes
parentf68fa8b45dfac545cfc4111d4e52804c86db68d3 (diff)
downloaddjango-9c19aff7c7561e3a82978a272ecdaad40dda5c00.tar.gz
Refs #33476 -- Reformatted code with Black.
Diffstat (limited to 'tests/datetimes')
-rw-r--r--tests/datetimes/models.py2
-rw-r--r--tests/datetimes/tests.py117
2 files changed, 73 insertions, 46 deletions
diff --git a/tests/datetimes/models.py b/tests/datetimes/models.py
index c8b64ae52a..9c8e5e6cd9 100644
--- a/tests/datetimes/models.py
+++ b/tests/datetimes/models.py
@@ -19,7 +19,7 @@ class Comment(models.Model):
approval_date = models.DateTimeField(null=True)
def __str__(self):
- return 'Comment to %s (%s)' % (self.article.title, self.pub_date)
+ return "Comment to %s (%s)" % (self.article.title, self.pub_date)
class Category(models.Model):
diff --git a/tests/datetimes/tests.py b/tests/datetimes/tests.py
index 95828f8e09..58bf3ac0dc 100644
--- a/tests/datetimes/tests.py
+++ b/tests/datetimes/tests.py
@@ -49,31 +49,36 @@ class DateTimesTests(TestCase):
c.articles.add(a1, a3)
self.assertSequenceEqual(
- Comment.objects.datetimes("article__pub_date", "year"), [
+ Comment.objects.datetimes("article__pub_date", "year"),
+ [
datetime.datetime(2005, 1, 1),
datetime.datetime(2010, 1, 1),
],
)
self.assertSequenceEqual(
- Comment.objects.datetimes("article__pub_date", "month"), [
+ Comment.objects.datetimes("article__pub_date", "month"),
+ [
datetime.datetime(2005, 7, 1),
datetime.datetime(2010, 7, 1),
],
)
self.assertSequenceEqual(
- Comment.objects.datetimes("article__pub_date", "week"), [
+ Comment.objects.datetimes("article__pub_date", "week"),
+ [
datetime.datetime(2005, 7, 25),
datetime.datetime(2010, 7, 26),
],
)
self.assertSequenceEqual(
- Comment.objects.datetimes("article__pub_date", "day"), [
+ Comment.objects.datetimes("article__pub_date", "day"),
+ [
datetime.datetime(2005, 7, 28),
datetime.datetime(2010, 7, 28),
],
)
self.assertSequenceEqual(
- Article.objects.datetimes("comments__pub_date", "day"), [
+ Article.objects.datetimes("comments__pub_date", "day"),
+ [
datetime.datetime(2005, 7, 28),
datetime.datetime(2005, 7, 29),
datetime.datetime(2005, 8, 29),
@@ -84,7 +89,8 @@ class DateTimesTests(TestCase):
Article.objects.datetimes("comments__approval_date", "day"), []
)
self.assertSequenceEqual(
- Category.objects.datetimes("articles__pub_date", "day"), [
+ Category.objects.datetimes("articles__pub_date", "day"),
+ [
datetime.datetime(2005, 7, 28),
],
)
@@ -93,17 +99,17 @@ class DateTimesTests(TestCase):
def test_21432(self):
now = timezone.localtime(timezone.now().replace(microsecond=0))
Article.objects.create(title="First one", pub_date=now)
- qs = Article.objects.datetimes('pub_date', 'second')
+ qs = Article.objects.datetimes("pub_date", "second")
self.assertEqual(qs[0], now)
- @unittest.skipUnless(pytz is not None, 'Test requires pytz')
+ @unittest.skipUnless(pytz is not None, "Test requires pytz")
@ignore_warnings(category=RemovedInDjango50Warning)
- @override_settings(USE_TZ=True, TIME_ZONE='UTC', USE_DEPRECATED_PYTZ=True)
+ @override_settings(USE_TZ=True, TIME_ZONE="UTC", USE_DEPRECATED_PYTZ=True)
def test_datetimes_ambiguous_and_invalid_times(self):
- sao = pytz.timezone('America/Sao_Paulo')
+ sao = pytz.timezone("America/Sao_Paulo")
utc = pytz.UTC
article = Article.objects.create(
- title='Article 1',
+ title="Article 1",
pub_date=utc.localize(datetime.datetime(2016, 2, 21, 1)),
)
Comment.objects.create(
@@ -111,24 +117,28 @@ class DateTimesTests(TestCase):
pub_date=utc.localize(datetime.datetime(2016, 10, 16, 13)),
)
with timezone.override(sao):
- with self.assertRaisesMessage(pytz.AmbiguousTimeError, '2016-02-20 23:00:00'):
- Article.objects.datetimes('pub_date', 'hour').get()
- with self.assertRaisesMessage(pytz.NonExistentTimeError, '2016-10-16 00:00:00'):
- Comment.objects.datetimes('pub_date', 'day').get()
+ with self.assertRaisesMessage(
+ pytz.AmbiguousTimeError, "2016-02-20 23:00:00"
+ ):
+ Article.objects.datetimes("pub_date", "hour").get()
+ with self.assertRaisesMessage(
+ pytz.NonExistentTimeError, "2016-10-16 00:00:00"
+ ):
+ Comment.objects.datetimes("pub_date", "day").get()
self.assertEqual(
- Article.objects.datetimes('pub_date', 'hour', is_dst=False).get().dst(),
+ Article.objects.datetimes("pub_date", "hour", is_dst=False).get().dst(),
datetime.timedelta(0),
)
self.assertEqual(
- Comment.objects.datetimes('pub_date', 'day', is_dst=False).get().dst(),
+ Comment.objects.datetimes("pub_date", "day", is_dst=False).get().dst(),
datetime.timedelta(0),
)
self.assertEqual(
- Article.objects.datetimes('pub_date', 'hour', is_dst=True).get().dst(),
+ Article.objects.datetimes("pub_date", "hour", is_dst=True).get().dst(),
datetime.timedelta(0, 3600),
)
self.assertEqual(
- Comment.objects.datetimes('pub_date', 'hour', is_dst=True).get().dst(),
+ Comment.objects.datetimes("pub_date", "hour", is_dst=True).get().dst(),
datetime.timedelta(0, 3600),
)
@@ -137,30 +147,34 @@ class DateTimesTests(TestCase):
datetime.datetime(2005, 7, 28, 12, 15),
datetime.datetime(2005, 7, 29, 2, 15),
datetime.datetime(2005, 7, 30, 5, 15),
- datetime.datetime(2005, 7, 31, 19, 15)]
+ datetime.datetime(2005, 7, 31, 19, 15),
+ ]
for i, pub_date in enumerate(pub_dates):
- Article(pub_date=pub_date, title='title #{}'.format(i)).save()
+ Article(pub_date=pub_date, title="title #{}".format(i)).save()
self.assertSequenceEqual(
- Article.objects.datetimes('pub_date', 'year'),
+ Article.objects.datetimes("pub_date", "year"),
[datetime.datetime(2005, 1, 1, 0, 0)],
)
self.assertSequenceEqual(
- Article.objects.datetimes('pub_date', 'month'),
+ Article.objects.datetimes("pub_date", "month"),
[datetime.datetime(2005, 7, 1, 0, 0)],
)
self.assertSequenceEqual(
- Article.objects.datetimes('pub_date', 'week'),
+ Article.objects.datetimes("pub_date", "week"),
[datetime.datetime(2005, 7, 25, 0, 0)],
)
- self.assertSequenceEqual(Article.objects.datetimes('pub_date', 'day'), [
- datetime.datetime(2005, 7, 28, 0, 0),
- datetime.datetime(2005, 7, 29, 0, 0),
- datetime.datetime(2005, 7, 30, 0, 0),
- datetime.datetime(2005, 7, 31, 0, 0),
- ])
self.assertSequenceEqual(
- Article.objects.datetimes('pub_date', 'day', order='ASC'),
+ Article.objects.datetimes("pub_date", "day"),
+ [
+ datetime.datetime(2005, 7, 28, 0, 0),
+ datetime.datetime(2005, 7, 29, 0, 0),
+ datetime.datetime(2005, 7, 30, 0, 0),
+ datetime.datetime(2005, 7, 31, 0, 0),
+ ],
+ )
+ self.assertSequenceEqual(
+ Article.objects.datetimes("pub_date", "day", order="ASC"),
[
datetime.datetime(2005, 7, 28, 0, 0),
datetime.datetime(2005, 7, 29, 0, 0),
@@ -169,7 +183,7 @@ class DateTimesTests(TestCase):
],
)
self.assertSequenceEqual(
- Article.objects.datetimes('pub_date', 'day', order='DESC'),
+ Article.objects.datetimes("pub_date", "day", order="DESC"),
[
datetime.datetime(2005, 7, 31, 0, 0),
datetime.datetime(2005, 7, 30, 0, 0),
@@ -183,29 +197,42 @@ class DateTimesTests(TestCase):
datetime.datetime(2005, 7, 28, 12, 15),
datetime.datetime(2005, 7, 29, 2, 15),
datetime.datetime(2005, 7, 30, 5, 15),
- datetime.datetime(2005, 7, 31, 19, 15)]
+ datetime.datetime(2005, 7, 31, 19, 15),
+ ]
for i, pub_date in enumerate(pub_dates):
- Article(pub_date=pub_date, title='title #{}'.format(i)).save()
+ Article(pub_date=pub_date, title="title #{}".format(i)).save()
# Use iterator() with datetimes() to return a generator that lazily
# requests each result one at a time, to save memory.
dates = []
with self.assertNumQueries(0):
- article_datetimes_iterator = Article.objects.datetimes('pub_date', 'day', order='DESC').iterator()
+ article_datetimes_iterator = Article.objects.datetimes(
+ "pub_date", "day", order="DESC"
+ ).iterator()
with self.assertNumQueries(1):
for article in article_datetimes_iterator:
dates.append(article)
- self.assertEqual(dates, [
- datetime.datetime(2005, 7, 31, 0, 0),
- datetime.datetime(2005, 7, 30, 0, 0),
- datetime.datetime(2005, 7, 29, 0, 0),
- datetime.datetime(2005, 7, 28, 0, 0)])
+ self.assertEqual(
+ dates,
+ [
+ datetime.datetime(2005, 7, 31, 0, 0),
+ datetime.datetime(2005, 7, 30, 0, 0),
+ datetime.datetime(2005, 7, 29, 0, 0),
+ datetime.datetime(2005, 7, 28, 0, 0),
+ ],
+ )
def test_datetimes_disallows_date_fields(self):
dt = datetime.datetime(2005, 7, 28, 12, 15)
- Article.objects.create(pub_date=dt, published_on=dt.date(), title="Don't put dates into datetime functions!")
- with self.assertRaisesMessage(ValueError, "Cannot truncate DateField 'published_on' to DateTimeField"):
- list(Article.objects.datetimes('published_on', 'second'))
+ Article.objects.create(
+ pub_date=dt,
+ published_on=dt.date(),
+ title="Don't put dates into datetime functions!",
+ )
+ with self.assertRaisesMessage(
+ ValueError, "Cannot truncate DateField 'published_on' to DateTimeField"
+ ):
+ list(Article.objects.datetimes("published_on", "second"))
def test_datetimes_fails_when_given_invalid_kind_argument(self):
msg = (
@@ -213,9 +240,9 @@ class DateTimesTests(TestCase):
"'minute', or 'second'."
)
with self.assertRaisesMessage(ValueError, msg):
- Article.objects.datetimes('pub_date', 'bad_kind')
+ Article.objects.datetimes("pub_date", "bad_kind")
def test_datetimes_fails_when_given_invalid_order_argument(self):
msg = "'order' must be either 'ASC' or 'DESC'."
with self.assertRaisesMessage(ValueError, msg):
- Article.objects.datetimes('pub_date', 'year', order='bad order')
+ Article.objects.datetimes("pub_date", "year", order="bad order")