fix: tests (#5)

* test: more examples for enum * test: use bounded dates
bdura · Apr 19, 2024 · d674c32 · d674c32
1 parent bbb117b
commit d674c32
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/tests/test_from_enum.py b/tests/test_from_enum.py
@@ -15,8 +15,8 @@ class Defcon(enum.Enum):
 
 EXAMPLES = [
     ("5", Defcon.FIVE),
-    # ("3", Defcon.THREE),
-    # ("1", Defcon.ONE),
+    ("3", Defcon.THREE),
+    ("1", Defcon.ONE),
 ]
 
 parser = from_enum(Defcon)

diff --git a/tests/test_from_stream.py b/tests/test_from_stream.py
@@ -22,8 +22,13 @@ def datetime_parser(stream: Stream[str]) -> datetime:
     return datetime(year, month, day)
 
 
-@given(st.datetimes())
-def test_generate(dt: datetime):
+@given(
+    st.datetimes(
+        min_value=datetime(1000, 1, 1),
+        max_value=datetime(9999, 12, 31),
+    )
+)
+def test_datetime_parser_from_stream(dt: datetime):
     dt = datetime(dt.year, dt.month, dt.day)
     text = dt.strftime("%Y-%m-%d")
     assert datetime_parser.parse(text) == dt