Timedelta deterministic hashing (#9213)

dask · Jun 24, 2022 · 0204e1a · 0204e1a
1 parent 0ee07e3
commit 0204e1a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/dask/base.py b/dask/base.py
@@ -949,6 +949,7 @@ def tokenize(*args, **kwargs):
         complex,
         type(Ellipsis),
         datetime.date,
+        datetime.timedelta,
     ),
     identity,
 )

diff --git a/dask/tests/test_base.py b/dask/tests/test_base.py
@@ -426,6 +426,11 @@ def test_tokenize_ordered_dict():
     assert tokenize(a) != tokenize(c)
 
 
+def test_tokenize_timedelta():
+    assert tokenize(datetime.timedelta(days=1)) == tokenize(datetime.timedelta(days=1))
+    assert tokenize(datetime.timedelta(days=1)) != tokenize(datetime.timedelta(days=2))
+
+
 @pytest.mark.parametrize("enum_type", [Enum, IntEnum, IntFlag, Flag])
 def test_tokenize_enum(enum_type):
     class Color(enum_type):