Fix patching not returning datetime subclasses

adamchainz · May 1, 2024 · 210f9d1 · 210f9d1
1 parent 9ced2bb
commit 210f9d1
Showing 1 changed file with 9 additions and 0 deletions.
diff --git 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()