Teach astroid about Hypothesis

Author: Zac-HD

ChangeLog

@@ -7,6 +7,8 @@ What's New in astroid 2.5.0?
 ============================
 Release Date: TBA
 
+* Added a brain for ``hypothesis.strategies.composite``
+
 * Added a brain for ``sqlalchemy.orm.session``
 
 * Separate string and bytes classes patching

astroid/brain/brain_hypothesis.py

@@ -0,0 +1,54 @@
+# Licensed under the LGPL: https://www.gnu.org/licenses/old-licenses/lgpl-2.1.en.html
+# For details: https://github.com/PyCQA/astroid/blob/master/COPYING.LESSER
+"""
+Astroid hook for the Hypothesis library.
+
+Without this hook pylint reports no-value-for-parameter for use of strategies
+defined using the `@hypothesis.strategies.composite` decorator.  For example:
+
+    from hypothesis import strategies as st
+
+    @st.composite
+    def a_strategy(draw):
+        return draw(st.integers())
+
+    a_strategy()
+
+"""
+
+import astroid
+
+COMPOSITE_NAMES = (
+    "st.composite",
+    "strategies.composite",
+    "hypothesis.strategies.composite",
+)
+
+
+def is_decorated_with_st_composite(node):
+    """Return True if a decorated node has @st.composite applied."""
+    if not node.decorators:
+        return False
+    for decorator_attribute in node.decorators.nodes:
+        if decorator_attribute.as_string() in COMPOSITE_NAMES:
+            return True
+    return False
+
+
+def remove_draw_parameter_from_composite_strategy(node):
+    """Given that the FunctionDef is decorated with @st.composite, remove the
+    first argument (`draw`) - it's always supplied by Hypothesis so we don't
+    need to emit the no-value-for-parameter lint.
+    """
+    if node.args.args[0].name == "draw":
+        del node.args.args[0]
+        del node.args.annotations[0]
+        del node.args.type_comment_args[0]
+    return node
+
+
+astroid.MANAGER.register_transform(
+    node_class=astroid.FunctionDef,
+    transform=remove_draw_parameter_from_composite_strategy,
+    predicate=is_decorated_with_st_composite,
+)