formating

Author: Tanguy-Boisset

src/neo4j/time/__init__.py

@@ -2019,6 +2019,12 @@ def __new__(
         return cls.combine(Date(year, month, day),
                            Time(hour, minute, second, nanosecond, tzinfo))
 
+    # SERIALIZATION #
+
+    def __reduce__(cls):
+        return (cls.__class__,
+                (cls.year, cls.month, cls.day, cls.hour, cls.minute, cls.second, cls.nanosecond, cls.tzinfo))
+
     # CLASS METHODS #
 
     @classmethod

tests/unit/common/time/test_datetime.py

@@ -19,6 +19,7 @@
 import copy
 import itertools
 import operator
+import pickle
 from datetime import (
     datetime,
     timedelta,
@@ -410,6 +411,11 @@ def test_datetime_deep_copy(self) -> None:
         assert d is not d2
         assert d == d2
 
+    def test_datetime_pickle(self) -> None:
+        expected = DateTime(2023, 12, 7, 12, 34, 56, 7890)
+        actual = pickle.loads(pickle.dumps(expected))
+        assert expected == actual
+
 
 def test_iso_format_with_time_zone_case_1() -> None:
     # python -m pytest tests/unit/time/test_datetime.py -s -v -k test_iso_format_with_time_zone_case_1