From 210f9d184316cffb102c34452f5bf40726d2c305 Mon Sep 17 00:00:00 2001 From: Adam Johnson Date: Wed, 1 May 2024 22:36:46 +0100 Subject: [PATCH] Fix patching not returning datetime subclasses --- tests/test_time_machine.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/tests/test_time_machine.py b/tests/test_time_machine.py index 55571484..027089d4 100644 --- a/tests/test_time_machine.py +++ b/tests/test_time_machine.py @@ -114,6 +114,15 @@ def test_datetime_now_arg(): ) +def test_datetime_now_subclass(): + class MyDatetime(dt.datetime): + pass + + with time_machine.travel(EPOCH): + now = MyDatetime.now() + assert isinstance(now, MyDatetime) + + def test_datetime_utcnow(): with time_machine.travel(EPOCH): now = dt.datetime.utcnow()