eFdZddlmZddlmZdZdefdZdZdedd fd Z y ) aV 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() )AstroidManager) FunctionDef) compositez st.compositezstrategies.compositezhypothesis.strategies.compositereturnc|jro|jjrY|jjdjdk(r3|jjD]}|j t vsyy)z:Return whether a decorated node has @st.composite applied.rdrawTF) decoratorsargsnamenodes as_stringCOMPOSITE_NAMES)nodedecorator_attributes @/usr/lib/python3/dist-packages/astroid/brain/brain_hypothesis.pyis_decorated_with_st_compositers` 499>>diinnQ.?.D.D.N#'??#8#8  ",,./A  c|jjd=|jjd=|jjd=|S)zGiven 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. r)r annotationstype_comment_args)rs r-remove_draw_parameter_from_composite_strategyr's= q a  ##A& KrmanagerNcD|jttty)N) node_class transform predicate)register_transformrrr)rs rregisterr2s ?0r) __doc__astroid.managerrastroid.nodes.scoped_nodesrrboolrrrrrr$s= +2Dnr