hypothesis 6.135.7__py3-none-any.whl → 6.135.9__py3-none-any.whl

hypothesis/extra/_patching.py

@@ -26,9 +26,11 @@ import re
 import sys
 import types
 from ast import literal_eval
+from collections.abc import Sequence
 from contextlib import suppress
 from datetime import date, datetime, timedelta, timezone
 from pathlib import Path
+from typing import Any, Optional
 
 import libcst as cst
 from libcst import matchers as m
@@ -43,11 +45,11 @@ try:
 except ImportError:
     black = None  # type: ignore
 
-HEADER = f"""\
+HEADER = """\
 From HEAD Mon Sep 17 00:00:00 2001
-From: Hypothesis {__version__} <no-reply@hypothesis.works>
-Date: {{when:%a, %d %b %Y %H:%M:%S}}
-Subject: [PATCH] {{msg}}
+From: {author}
+Date: {when:%a, %d %b %Y %H:%M:%S}
+Subject: [PATCH] {msg}
 
 ---
 """
@@ -56,7 +58,7 @@ _space_only_re = re.compile("^ +$", re.MULTILINE)
 _leading_space_re = re.compile("(^[ ]*)(?:[^ \n])", re.MULTILINE)
 
 
-def dedent(text):
+def dedent(text: str) -> tuple[str, str]:
     # Simplified textwrap.dedent, for valid Python source code only
     text = _space_only_re.sub("", text)
     prefix = min(_leading_space_re.findall(text), key=len)
@@ -70,7 +72,14 @@ def indent(text: str, prefix: str) -> str:
 class AddExamplesCodemod(VisitorBasedCodemodCommand):
     DESCRIPTION = "Add explicit examples to failing tests."
 
-    def __init__(self, context, fn_examples, strip_via=(), dec="example", width=88):
+    def __init__(
+        self,
+        context: CodemodContext,
+        fn_examples: dict[str, list[tuple[cst.Call, str]]],
+        strip_via: tuple[str, ...] = (),
+        decorator: str = "example",
+        width: int = 88,
+    ):
         """Add @example() decorator(s) for failing test(s).
 
         `code` is the source code of the module where the test functions are defined.
@@ -79,9 +88,11 @@ class AddExamplesCodemod(VisitorBasedCodemodCommand):
         assert fn_examples, "This codemod does nothing without fn_examples."
         super().__init__(context)
 
-        self.decorator_func = cst.parse_expression(dec)
+        self.decorator_func = cst.parse_expression(decorator)
         self.line_length = width
-        value_in_strip_via = m.MatchIfTrue(lambda x: literal_eval(x.value) in strip_via)
+        value_in_strip_via: Any = m.MatchIfTrue(
+            lambda x: literal_eval(x.value) in strip_via
+        )
         self.strip_matching = m.Call(
             m.Attribute(m.Call(), m.Name("via")),
             [m.Arg(m.SimpleString() & value_in_strip_via)],
@@ -89,11 +100,17 @@ class AddExamplesCodemod(VisitorBasedCodemodCommand):
 
         # Codemod the failing examples to Call nodes usable as decorators
         self.fn_examples = {
-            k: tuple(d for x in nodes if (d := self.__call_node_to_example_dec(*x)))
+            k: tuple(
+                d
+                for (node, via) in nodes
+                if (d := self.__call_node_to_example_dec(node, via))
+            )
             for k, nodes in fn_examples.items()
         }
 
-    def __call_node_to_example_dec(self, node, via):
+    def __call_node_to_example_dec(
+        self, node: cst.Call, via: str
+    ) -> Optional[cst.Decorator]:
         # If we have black installed, remove trailing comma, _unless_ there's a comment
         node = node.with_changes(
             func=self.decorator_func,
@@ -112,7 +129,7 @@ class AddExamplesCodemod(VisitorBasedCodemodCommand):
                 else node.args
             ),
         )
-        via = cst.Call(
+        via: cst.BaseExpression = cst.Call(
             func=cst.Attribute(node, cst.Name("via")),
             args=[cst.Arg(cst.SimpleString(repr(via)))],
         )
@@ -127,7 +144,9 @@ class AddExamplesCodemod(VisitorBasedCodemodCommand):
             via = cst.parse_expression(pretty.strip())
         return cst.Decorator(via)
 
-    def leave_FunctionDef(self, _, updated_node):
+    def leave_FunctionDef(
+        self, _original_node: cst.FunctionDef, updated_node: cst.FunctionDef
+    ) -> cst.FunctionDef:
         return updated_node.with_changes(
             # TODO: improve logic for where in the list to insert this decorator
             decorators=tuple(
@@ -140,30 +159,59 @@ class AddExamplesCodemod(VisitorBasedCodemodCommand):
         )
 
 
-def get_patch_for(func, failing_examples, *, strip_via=()):
-    # Skip this if we're unable to find the location or source of this function.
+def get_patch_for(
+    func: Any,
+    examples: Sequence[tuple[str, str]],
+    *,
+    strip_via: tuple[str, ...] = (),
+) -> Optional[tuple[str, str, str]]:
+    # Skip this if we're unable to find the location of this function.
     try:
         module = sys.modules[func.__module__]
-        fname = Path(module.__file__).relative_to(Path.cwd())
-        before = inspect.getsource(func)
+        file_path = Path(module.__file__)  # type: ignore
     except Exception:
         return None
 
+    fname = (
+        file_path.relative_to(Path.cwd())
+        if file_path.is_relative_to(Path.cwd())
+        else file_path
+    )
+    patch = _get_patch_for(
+        func, examples, strip_via=strip_via, namespace=module.__dict__
+    )
+    if patch is None:
+        return None
+
+    (before, after) = patch
+    return (str(fname), before, after)
+
+
+# split out for easier testing of patches in hypofuzz, where the function to
+# apply the patch to may not be loaded in sys.modules.
+def _get_patch_for(
+    func: Any,
+    examples: Sequence[tuple[str, str]],
+    *,
+    strip_via: tuple[str, ...] = (),
+    namespace: dict[str, Any],
+) -> Optional[tuple[str, str]]:
+    try:
+        before = inspect.getsource(func)
+    except Exception:  # pragma: no cover
+        return None
+
     modules_in_test_scope = sorted(
-        (
-            (k, v)
-            for (k, v) in module.__dict__.items()
-            if isinstance(v, types.ModuleType)
-        ),
+        ((k, v) for (k, v) in namespace.items() if isinstance(v, types.ModuleType)),
         key=lambda kv: len(kv[1].__name__),
     )
 
     # The printed examples might include object reprs which are invalid syntax,
     # so we parse here and skip over those.  If _none_ are valid, there's no patch.
-    call_nodes = []
-    for ex, via in set(failing_examples):
+    call_nodes: list[tuple[cst.Call, str]] = []
+    for ex, via in set(examples):
         with suppress(Exception):
-            node = cst.parse_module(ex)
+            node: Any = cst.parse_module(ex)
             the_call = node.body[0].body[0].value
             assert isinstance(the_call, cst.Call), the_call
             # Check for st.data(), which doesn't support explicit examples
@@ -180,7 +228,7 @@ def get_patch_for(func, failing_examples, *, strip_via=()):
                     isinstance(anode, ast.Name)
                     and isinstance(anode.ctx, ast.Load)
                     and anode.id not in names
-                    and anode.id not in module.__dict__
+                    and anode.id not in namespace
                 ):
                     for k, v in modules_in_test_scope:
                         if anode.id in v.__dict__:
@@ -194,14 +242,15 @@ def get_patch_for(func, failing_examples, *, strip_via=()):
             with suppress(Exception):
                 wrapper = cst.metadata.MetadataWrapper(node)
                 kwarg_names = {
-                    a.keyword for a in m.findall(wrapper, m.Arg(keyword=m.Name()))
+                    node.keyword  # type: ignore
+                    for node in m.findall(wrapper, m.Arg(keyword=m.Name()))
                 }
                 node = m.replace(
                     wrapper,
                     m.Name(value=m.MatchIfTrue(names.__contains__))
                     & m.MatchMetadata(ExpressionContextProvider, ExpressionContext.LOAD)
-                    & m.MatchIfTrue(lambda n, k=kwarg_names: n not in k),
-                    replacement=lambda node, _, ns=names: ns[node.value],
+                    & m.MatchIfTrue(lambda n, k=kwarg_names: n not in k),  # type: ignore
+                    replacement=lambda node, _, ns=names: ns[node.value],  # type: ignore
                 )
             node = node.body[0].body[0].value
             assert isinstance(node, cst.Call), node
@@ -211,8 +260,8 @@ def get_patch_for(func, failing_examples, *, strip_via=()):
         return None
 
     if (
-        module.__dict__.get("hypothesis") is sys.modules["hypothesis"]
-        and "given" not in module.__dict__  # more reliably present than `example`
+        namespace.get("hypothesis") is sys.modules["hypothesis"]
+        and "given" not in namespace  # more reliably present than `example`
     ):
         decorator_func = "hypothesis.example"
     else:
@@ -229,22 +278,28 @@ def get_patch_for(func, failing_examples, *, strip_via=()):
         CodemodContext(),
         fn_examples={func.__name__: call_nodes},
         strip_via=strip_via,
-        dec=decorator_func,
+        decorator=decorator_func,
         width=88 - len(prefix),  # to match Black's default formatting
     ).transform_module(node)
-    return (str(fname), before, indent(after.code, prefix=prefix))
+    return (before, indent(after.code, prefix=prefix))
 
 
-def make_patch(triples, *, msg="Hypothesis: add explicit examples", when=None):
+def make_patch(
+    triples: Sequence[tuple[str, str, str]],
+    *,
+    msg: str = "Hypothesis: add explicit examples",
+    when: Optional[datetime] = None,
+    author: str = f"Hypothesis {__version__} <no-reply@hypothesis.works>",
+) -> str:
     """Create a patch for (fname, before, after) triples."""
     assert triples, "attempted to create empty patch"
     when = when or datetime.now(tz=timezone.utc)
 
-    by_fname = {}
+    by_fname: dict[Path, list[tuple[str, str]]] = {}
     for fname, before, after in triples:
         by_fname.setdefault(Path(fname), []).append((before, after))
 
-    diffs = [HEADER.format(msg=msg, when=when)]
+    diffs = [HEADER.format(msg=msg, when=when, author=author)]
     for fname, changes in sorted(by_fname.items()):
         source_before = source_after = fname.read_text(encoding="utf-8")
         for before, after in changes:

hypothesis/internal/constants_ast.py

@@ -192,6 +192,10 @@ def _constants_from_source(source: Union[str, bytes], *, limit: bool) -> Constan
     return visitor.constants
 
 
+def _constants_file_str(constants: Constants) -> str:
+    return str(sorted(constants, key=lambda v: (str(type(v)), v)))
+
+
 @lru_cache(4096)
 def constants_from_module(module: ModuleType, *, limit: bool = True) -> Constants:
     try:
@@ -236,7 +240,7 @@ def constants_from_module(module: ModuleType, *, limit: bool = True) -> Constant
             # somewhat arbitrary sort order. The cache file doesn't *have* to be
             # stable... but it is aesthetically pleasing, and means we could rely
             # on it in the future!
-            + str(sorted(constants, key=lambda v: (str(type(v)), v))),
+            + _constants_file_str(constants),
             encoding="utf-8",
         )
     except Exception:  # pragma: no cover

hypothesis/version.py

@@ -8,5 +8,5 @@
 # v. 2.0. If a copy of the MPL was not distributed with this file, You can
 # obtain one at https://mozilla.org/MPL/2.0/.
 
-__version_info__ = (6, 135, 7)
+__version_info__ = (6, 135, 9)
 __version__ = ".".join(map(str, __version_info__))

{hypothesis-6.135.7.dist-info → hypothesis-6.135.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hypothesis
-Version: 6.135.7
+Version: 6.135.9
 Summary: A library for property-based testing
 Author-email: "David R. MacIver and Zac Hatfield-Dodds" <david@drmaciver.com>
 License-Expression: MPL-2.0

{hypothesis-6.135.7.dist-info → hypothesis-6.135.9.dist-info}/RECORD

@@ -14,10 +14,10 @@ hypothesis/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hypothesis/reporting.py,sha256=f-jhl1JfAi5_tG8dsUd2qDjGcPdvxEzfF6hXmpTFQ1g,1761
 hypothesis/stateful.py,sha256=vQ8wDO7YW-8nGlBGLfR9ariwmRCS9jxJy-Ky3swXjDE,42861
 hypothesis/statistics.py,sha256=kZ5mc0fAg7gnSO6EmDo82fyz8DYhIiJ_mHe7srxOeQ0,5438
-hypothesis/version.py,sha256=d-zCITTEEu8cpXH2B5XV2DOcDRJYHRLHDNA5I83QjVY,498
+hypothesis/version.py,sha256=2cgiH3RVZ2YHrP3mS9FXI4dZXTKhr9HPeF4V-9yoL_A,498
 hypothesis/extra/__init__.py,sha256=gx4ENVDkrzBxy5Lv3Iyfs3tvMGdWMbiHfi95B7t61CY,415
 hypothesis/extra/_array_helpers.py,sha256=PLmFckBfQpzQ4Q3dFJQqMmrbm7Qdvqxf1t9LHDCuSp0,27627
-hypothesis/extra/_patching.py,sha256=pLgtFBgwDEotw6Pu8ZfJbkE5Qk5F1H3IH5EPUo0Ngm4,10869
+hypothesis/extra/_patching.py,sha256=A5s5EAf81itr--w4SAFyzuecSZm4eT397jM7BvbnQXU,12385
 hypothesis/extra/array_api.py,sha256=bv_wisQS56wvwg-LiC26DxvzmPxcNLzlbzmltFWZ_Jo,42609
 hypothesis/extra/cli.py,sha256=HOH_BGosyUvS4yNDsWF4Dfe672tEFGT4BBPBuZamm1s,12885
 hypothesis/extra/codemods.py,sha256=i_iM5aSnrNjSZ_uNW40yShxXo7qaXuCrDHy8OznOa7o,11224
@@ -39,7 +39,7 @@ hypothesis/internal/cache.py,sha256=5GJ92S7HNjefuiOOSUchp3IYw1XxLVRmYoFHEoL7eVk,
 hypothesis/internal/cathetus.py,sha256=q6t16mT2jzlJmBddhRlqC2wg_w7FggVDRAVkzs04W-U,2258
 hypothesis/internal/charmap.py,sha256=dIsRCxa6r1r-rF_NUy04Z-afFa2b6fCFEjWTMYsAgXY,11781
 hypothesis/internal/compat.py,sha256=CexhnEVndODX3E-iJmBzt_svqLmADF_6vWkd7lXLRQI,10970
-hypothesis/internal/constants_ast.py,sha256=to4zxaUpA4JU5iPXy2pOb38dWG_tibu6UZkd2Vrq-fA,10091
+hypothesis/internal/constants_ast.py,sha256=VcItL6O6LbNd-NpunBuonKE9rLNVVxNlOfTDY8Lqci0,10189
 hypothesis/internal/coverage.py,sha256=u2ALnaYgwd2jSkZkA5aQYA1uMnk7gwaasMJbq_l3iIg,3380
 hypothesis/internal/detection.py,sha256=v_zf0GYsnfJKQMNw78qYv61Dh-YaXXfgqKpVIOsBvfw,1228
 hypothesis/internal/entropy.py,sha256=wgNeddIM0I3SD1yydz12Lsop6Di5vsle4e9u5fWAxIs,8014
@@ -105,9 +105,9 @@ hypothesis/utils/terminal.py,sha256=IxGYDGaE4R3b_vMfz5buWbN18XH5qVP4IxqAgNAU5as,
 hypothesis/vendor/__init__.py,sha256=gx4ENVDkrzBxy5Lv3Iyfs3tvMGdWMbiHfi95B7t61CY,415
 hypothesis/vendor/pretty.py,sha256=WEZC-UV-QQgCjUf2Iz1WWaWnbgT7Hc3s-GWEVxq-Qz0,36114
 hypothesis/vendor/tlds-alpha-by-domain.txt,sha256=W9hYvpu2BMmNgE-SfPp8-GTzEVjw0HJUviqlvHwpZu8,9588
-hypothesis-6.135.7.dist-info/licenses/LICENSE.txt,sha256=rIkDe6xjVQZE3OjPMsZ2Xl-rncGhzpS4n4qAXzQaZ1A,17141
-hypothesis-6.135.7.dist-info/METADATA,sha256=gZG_kjSYwAYhCC7U6l7rU90A1aN206QkGMDvIzbft9U,5637
-hypothesis-6.135.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-hypothesis-6.135.7.dist-info/entry_points.txt,sha256=JDoUs9w1bYme7aG_eJ1cCtstRTWD71BzG8iRi-G2eHE,113
-hypothesis-6.135.7.dist-info/top_level.txt,sha256=ReGreaueiJ4d1I2kEiig_CLeA0sD4QCQ4qk_8kH1oDc,81
-hypothesis-6.135.7.dist-info/RECORD,,
+hypothesis-6.135.9.dist-info/licenses/LICENSE.txt,sha256=rIkDe6xjVQZE3OjPMsZ2Xl-rncGhzpS4n4qAXzQaZ1A,17141
+hypothesis-6.135.9.dist-info/METADATA,sha256=Y5tNxfqmY2oL2NkcTrxP3ytDT-iZnoZT5kfz1XIUKa8,5637
+hypothesis-6.135.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+hypothesis-6.135.9.dist-info/entry_points.txt,sha256=JDoUs9w1bYme7aG_eJ1cCtstRTWD71BzG8iRi-G2eHE,113
+hypothesis-6.135.9.dist-info/top_level.txt,sha256=ReGreaueiJ4d1I2kEiig_CLeA0sD4QCQ4qk_8kH1oDc,81
+hypothesis-6.135.9.dist-info/RECORD,,