test: put deadline increase in the right place

AutoResearch · Nov 22, 2023 · b0f89b4 · b0f89b4
1 parent 4f6158e
commit b0f89b4
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/tests/test_serializer.py b/tests/test_serializer.py
@@ -6,7 +6,6 @@
 from collections import namedtuple
 
 import hypothesis.strategies as st
-from hypothesis import settings
 
 logger = logging.getLogger(__name__)
 
@@ -68,7 +67,6 @@ def _load_dump_via_disk(o):
     return _load_dump_via_disk
 
 
-@settings(deadline=500)  # some serializers are slow
 @st.composite
 def serializer_dump_load_strategy(draw):
     """Strategy returns a function which dumps an object and reloads it via a supported method."""

diff --git a/tests/test_state.py b/tests/test_state.py
@@ -11,7 +11,7 @@
 logger = logging.getLogger(__name__)
 
 
-@settings(suppress_health_check={HealthCheck.too_slow})
+@settings(suppress_health_check={HealthCheck.too_slow}, deadline=500)
 @given(standard_state_dataclass_strategy(), serializer_dump_load_strategy())
 def test_state_serialize_deserialize(o: StandardStateDataClass, dump_load):
     o_loaded = dump_load(o)