From 0204e1a6c4cefe965e1724a35b17db1787fd63d3 Mon Sep 17 00:00:00 2001 From: Fabien Aulaire <306648+faulaire@users.noreply.github.com> Date: Fri, 24 Jun 2022 16:47:13 +0200 Subject: [PATCH] Timedelta deterministic hashing (#9213) --- dask/base.py | 1 + dask/tests/test_base.py | 5 +++++ 2 files changed, 6 insertions(+) diff --git a/dask/base.py b/dask/base.py index 60b0f711ef5..91a8dbab215 100644 --- 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 index 709c745e933..69bf154359e 100644 --- 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):