From 7638e12c40e011be1386b14ea65414c3a2c445ce Mon Sep 17 00:00:00 2001 From: Steven Silvester Date: Mon, 14 Oct 2024 07:40:55 -0500 Subject: [PATCH] fix utc now handling --- bindings/python/test/test_builders.py | 4 ++-- bindings/python/test/test_datetime.py | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/bindings/python/test/test_builders.py b/bindings/python/test/test_builders.py index 91867763..ab16c4bb 100644 --- a/bindings/python/test/test_builders.py +++ b/bindings/python/test/test_builders.py @@ -12,7 +12,7 @@ # See the License for the specific language governing permissions and # limitations under the License. import calendar -from datetime import UTC, date, datetime, timedelta +from datetime import date, datetime, timedelta, timezone from unittest import TestCase from bson import Binary, Code, Decimal128, ObjectId @@ -86,7 +86,7 @@ def test_simple(self): self.maxDiff = None builder = DatetimeBuilder(dtype=timestamp("ms")) - datetimes = [datetime.now(UTC) + timedelta(days=k * 100) for k in range(5)] + datetimes = [datetime.now(timezone.UTC) + timedelta(days=k * 100) for k in range(5)] builder.append(self._datetime_to_millis(datetimes[0])) builder.append_values([self._datetime_to_millis(k) for k in datetimes[1:]]) builder.append_null() diff --git a/bindings/python/test/test_datetime.py b/bindings/python/test/test_datetime.py index 68e4ed32..3cd4638f 100644 --- a/bindings/python/test/test_datetime.py +++ b/bindings/python/test/test_datetime.py @@ -12,7 +12,7 @@ # See the License for the specific language governing permissions and # limitations under the License. import unittest -from datetime import UTC, datetime, timedelta +from datetime import datetime, timedelta, timezone from test import client_context import pytz @@ -36,8 +36,8 @@ def setUp(self): self.coll.drop() self.coll.insert_many( [ - {"_id": 1, "data": datetime.now(UTC) + timedelta(milliseconds=10)}, - {"_id": 2, "data": datetime.now(UTC) + timedelta(milliseconds=25)}, + {"_id": 1, "data": datetime.now(timezone.UTC) + timedelta(milliseconds=10)}, + {"_id": 2, "data": datetime.now(timezone.UTC) + timedelta(milliseconds=25)}, ] ) self.expected_times = []