Merge pull request #607 from eth-brownie/feat-address-strategy-length

Address strategy length

brownie/test/strategies.py

@@ -76,8 +76,10 @@ def _decimal_strategy(
 
 
 @_exclude_filter
-def _address_strategy() -> SearchStrategy:
-    return _DeferredStrategyRepr(lambda: st.sampled_from(list(network.accounts)), "accounts")
+def _address_strategy(length: Optional[int] = None) -> SearchStrategy:
+    return _DeferredStrategyRepr(
+        lambda: st.sampled_from(list(network.accounts)[:length]), "accounts"
+    )
 
 
 @_exclude_filter

docs/tests-hypothesis-property.rst

@@ -130,6 +130,7 @@ Address
 
 Optional keyword arguments:
 
+    * ``length``: The number of :func:`Account <brownie.network.account.Account>` objects to include in the strategy. If the :func:`Accounts <brownie.network.account.Accounts>` container holds less than this number of objects, the entire container is used.
     * ``excludes``: An object, iterable or callable used to filter strategy results.
 
 .. code-block:: python

tests/test/strategies/test_address.py

@@ -17,5 +17,10 @@ def test_given(accounts, value):
     assert isinstance(value, Account)
 
 
+@given(value=strategy("address", length=3))
+def test_length(accounts, value):
+    assert accounts.index(value) < 3
+
+
 def test_repr():
     assert repr(strategy("address")) == "sampled_from(accounts)"