hypothesis: accept container strategy in check_all_laws.

I'm calling it `container_strategy` to leave room for future extensions
that would allow configuring the pure function strategy, etc.

Author: pradeep90

returns/contrib/hypothesis/laws.py

@@ -23,14 +23,15 @@ class Settings(NamedTuple):
 
     settings_kwargs: dict[str, Any]
     use_init: bool
-    strategy: StrategyFactory | None = None
+    container_strategy: StrategyFactory | None
 
 
 def check_all_laws(
     container_type: type[Lawful],
     *,
     settings_kwargs: dict[str, Any] | None = None,
     use_init: bool = False,
+    container_strategy: StrategyFactory | None = None,
 ) -> None:
     """
     Function to check all defined mathematical laws in a specified container.
@@ -61,6 +62,7 @@ def check_all_laws(
     settings = Settings(
         settings_kwargs or {},
         use_init,
+        container_strategy,
     )
 
     for interface, laws in container_type.laws().items():
@@ -227,8 +229,8 @@ def _strategy_for_container(
 ) -> StrategyFactory:
     return (
         strategy_from_container(container_type, use_init=settings.use_init)
-        if settings.strategy is None
-        else settings.strategy
+        if settings.container_strategy is None
+        else settings.container_strategy
     )
 
 

returns/contrib/hypothesis/type_resolver.py

@@ -20,7 +20,7 @@ def strategies_for_types(
     Temporarily register strategies with `hypothesis`.
 
     Within this context, `hypothesis` will generate data for `MyType`
-    using `mapping[MyType]`, if possible. Otherwise, it will continue to
+    using `mapping[MyType]`, if available. Otherwise, it will continue to
     use the globally registered strategy for `MyType`.
 
     NOTE: This manually adds and removes strategies from an internal data

tests/test_contrib/test_hypothesis/test_laws/test_user_specified_strategy.py

@@ -0,0 +1,11 @@
+from hypothesis import strategies as st
+from test_hypothesis.test_laws import test_custom_type_applicative
+
+from returns.contrib.hypothesis.laws import check_all_laws
+
+container_type = test_custom_type_applicative._Wrapper  # noqa: SLF001
+
+check_all_laws(
+    container_type,
+    container_strategy=st.builds(container_type, st.integers()),
+)

tests/test_contrib/test_hypothesis/test_type_resolution.py

@@ -131,7 +131,10 @@ def test_register_container_with_no_strategy() -> None:
     container_type = Result
 
     with register_container(
-        container_type, settings=Settings(settings_kwargs={}, use_init=False)
+        container_type,
+        settings=Settings(
+            settings_kwargs={}, use_init=False, container_strategy=None
+        ),
     ):
         strategy_factory = look_up_strategy(container_type)
 
@@ -151,7 +154,9 @@ def test_register_container_with_strategy() -> None:
         }),
         register_container(
             container_type,
-            settings=Settings(settings_kwargs={}, use_init=False),
+            settings=Settings(
+                settings_kwargs={}, use_init=False, container_strategy=None
+            ),
         ),
     ):
         strategy_factory = look_up_strategy(container_type)
@@ -171,7 +176,7 @@ def test_register_container_with_setting() -> None:
         settings=Settings(
             settings_kwargs={},
             use_init=False,
-            strategy=st.builds(Success, st.integers()),
+            container_strategy=st.builds(Success, st.integers()),
         ),
     ):
         strategy_factory = look_up_strategy(container_type)
@@ -186,19 +191,14 @@ def test_interface_strategies() -> None:
     """Check that ancestor interfaces get resolved to the concrete container."""
     container_type = test_custom_type_applicative._Wrapper  # noqa: SLF001
 
-    strategy_factories_before = _interface_factories(container_type)
-
     with interface_strategies(
-        container_type, settings=Settings(settings_kwargs={}, use_init=False)
+        container_type,
+        settings=Settings(
+            settings_kwargs={}, use_init=False, container_strategy=None
+        ),
     ):
         strategy_factories_inside = _interface_factories(container_type)
 
-    strategy_factories_after = _interface_factories(container_type)
-
-    assert _strategy_strings(strategy_factories_before, container_type) == [
-        'None',
-        'None',
-    ]
     assert _strategy_strings(strategy_factories_inside, container_type) == [
         "builds(from_value, shared(sampled_from([<class 'NoneType'>,"
         " <class 'bool'>, <class 'int'>, <class 'float'>, <class 'str'>,"
@@ -207,42 +207,26 @@ def test_interface_strategies() -> None:
         " <class 'bool'>, <class 'int'>, <class 'float'>, <class 'str'>,"
         " <class 'bytes'>]), key='typevar=~_FirstType').flatmap(from_type))",
     ]
-    assert _strategy_strings(strategy_factories_after, container_type) == [
-        'None',
-        'None',
-    ]
 
 
 def test_interface_strategies_with_settings() -> None:
     """Check that we prefer the strategy in the settings."""
     container_type = test_custom_type_applicative._Wrapper  # noqa: SLF001
 
-    strategy_factories_before = _interface_factories(container_type)
-
     with interface_strategies(
         container_type,
         settings=Settings(
             settings_kwargs={},
             use_init=False,
-            strategy=st.builds(container_type, st.integers()),
+            container_strategy=st.builds(container_type, st.integers()),
         ),
     ):
         strategy_factories_inside = _interface_factories(container_type)
 
-    strategy_factories_after = _interface_factories(container_type)
-
-    assert _strategy_strings(strategy_factories_before, container_type) == [
-        'None',
-        'None',
-    ]
     assert _strategy_strings(strategy_factories_inside, container_type) == [
         'builds(_Wrapper, integers())',
         'builds(_Wrapper, integers())',
     ]
-    assert _strategy_strings(strategy_factories_after, container_type) == [
-        'None',
-        'None',
-    ]
 
 
 def _interface_factories(type_: type[Lawful]) -> list[StrategyFactory | None]: