Add future and past date hypothesis strategies (#5850)

pydantic · Jun 1, 2023 · 0813a03 · 0813a03
1 parent ca10302
commit 0813a03
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/changes/5850-bschoenmaeckers.md b/changes/5850-bschoenmaeckers.md
@@ -0,0 +1 @@
+Add future and past date hypothesis strategies.
diff --git a/pydantic/_hypothesis_plugin.py b/pydantic/_hypothesis_plugin.py
@@ -168,6 +168,11 @@ def add_luhn_digit(card_number: str) -> str:
 st.register_type_strategy(pydantic.StrictStr, st.text())
 
 
+# FutureDate, PastDate
+st.register_type_strategy(pydantic.FutureDate, st.dates(min_value=datetime.date.today() + datetime.timedelta(days=1)))
+st.register_type_strategy(pydantic.PastDate, st.dates(max_value=datetime.date.today() - datetime.timedelta(days=1)))
+
+
 # Constrained-type resolver functions
 #
 # For these ones, we actually want to inspect the type in order to work out a

diff --git a/tests/test_hypothesis_plugin.py b/tests/test_hypothesis_plugin.py
@@ -86,6 +86,8 @@ class ConstrainedNumbersModel(pydantic.BaseModel):
     class ConstrainedDateModel(pydantic.BaseModel):
         condatet: pydantic.condate(gt=date(1980, 1, 1), lt=date(2180, 12, 31))
         condatee: pydantic.condate(ge=date(1980, 1, 1), le=date(2180, 12, 31))
+        future: pydantic.FutureDate
+        past: pydantic.PastDate
 
     yield from (
         MiscModel,