Update tests/test_param.py

2025-07-17 10:07:31 +02:00
parent 466b07d34a
commit f6bb324a66
1 changed files with 12 additions and 0 deletions
@@ -30,3 +30,15 @@ def test_multiple_positional_args(a, b):
 ])
 def test_non_literal_with_id(data):
    pass
+
+from datetime import datetime, timedelta
+
+testdata = [
+    (datetime(2001, 12, 12), datetime(2001, 12, 11), timedelta(1)),
+    (datetime(2001, 12, 11), datetime(2001, 12, 12), timedelta(-1)),
+]
+
+@pytest.mark.parametrize("a,b,expected", testdata, ids=["forward", "backward"])
+def test_timedistance_v1(a, b, expected):
+    diff = a - b
+    assert diff == expected