diff options
author | django-bot <ops@djangoproject.com> | 2022-02-03 20:24:19 +0100 |
---|---|---|
committer | Mariusz Felisiak <felisiak.mariusz@gmail.com> | 2022-02-07 20:37:05 +0100 |
commit | 9c19aff7c7561e3a82978a272ecdaad40dda5c00 (patch) | |
tree | f0506b668a013d0063e5fba3dbf4863b466713ba /tests/datetimes | |
parent | f68fa8b45dfac545cfc4111d4e52804c86db68d3 (diff) | |
download | django-9c19aff7c7561e3a82978a272ecdaad40dda5c00.tar.gz |
Refs #33476 -- Reformatted code with Black.
Diffstat (limited to 'tests/datetimes')
-rw-r--r-- | tests/datetimes/models.py | 2 | ||||
-rw-r--r-- | tests/datetimes/tests.py | 117 |
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") |