diff options
| -rw-r--r-- | tests/migrations/test_writer.py | 16 |
1 files changed, 9 insertions, 7 deletions
diff --git a/tests/migrations/test_writer.py b/tests/migrations/test_writer.py index 5635dc62f0..96b2140089 100644 --- a/tests/migrations/test_writer.py +++ b/tests/migrations/test_writer.py @@ -492,13 +492,15 @@ class WriterTests(SimpleTestCase): datetime.datetime(2014, 1, 1, 1, 1), ("datetime.datetime(2014, 1, 1, 1, 1)", {'import datetime'}) ) - self.assertSerializedResultEqual( - datetime.datetime(2012, 1, 1, 1, 1, tzinfo=utc), - ( - "datetime.datetime(2012, 1, 1, 1, 1, tzinfo=utc)", - {'import datetime', 'from django.utils.timezone import utc'}, - ) - ) + for tzinfo in (utc, datetime.timezone.utc): + with self.subTest(tzinfo=tzinfo): + self.assertSerializedResultEqual( + datetime.datetime(2012, 1, 1, 1, 1, tzinfo=tzinfo), + ( + "datetime.datetime(2012, 1, 1, 1, 1, tzinfo=utc)", + {'import datetime', 'from django.utils.timezone import utc'}, + ) + ) def test_serialize_fields(self): self.assertSerializedFieldEqual(models.CharField(max_length=255)) |