support older pythons also

gijzelaerr · Nov 2, 2024 · 898cc76 · 898cc76
1 parent f7bc398
commit 898cc76
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tests/test_client.py b/tests/test_client.py
@@ -18,7 +18,7 @@
     pointer,
     Array,
 )
-from datetime import datetime, timedelta, UTC
+from datetime import datetime, timedelta, timezone
 from multiprocessing import Process
 from unittest import mock
 from typing import cast as typing_cast
@@ -804,10 +804,12 @@ def test_get_pg_block_info(self) -> None:
         self.assertEqual(99, block_info.BlkNumber)
         self.assertEqual(2752512, block_info.SBBLength)
         self.assertEqual(
-            bytes((datetime(2019, 6, 27, tzinfo=UTC).astimezone().strftime("%Y/%m/%d")), encoding="utf-8"), block_info.CodeDate
+            bytes((datetime(2019, 6, 27, tzinfo=timezone.utc).astimezone().strftime("%Y/%m/%d")), encoding="utf-8"),
+            block_info.CodeDate,
         )
         self.assertEqual(
-            bytes((datetime(2019, 6, 27, tzinfo=UTC).astimezone().strftime("%Y/%m/%d")), encoding="utf-8"), block_info.IntfDate
+            bytes((datetime(2019, 6, 27, tzinfo=timezone.utc).astimezone().strftime("%Y/%m/%d")), encoding="utf-8"),
+            block_info.IntfDate,
         )
 
     def test_iso_exchange_buffer(self) -> None: