File: test_modify_inner_test.py

package info (click to toggle)
python-hypothesis 6.138.0-1
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 15,272 kB
  • sloc: python: 62,853; ruby: 1,107; sh: 253; makefile: 41; javascript: 6
file content (104 lines) | stat: -rw-r--r-- 2,814 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
# This file is part of Hypothesis, which may be found at
# https://github.com/HypothesisWorks/hypothesis/
#
# Copyright the Hypothesis Authors.
# Individual contributors are listed in AUTHORS.rst and the git log.
#
# This Source Code Form is subject to the terms of the Mozilla Public License,
# 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/.

from functools import wraps

import pytest

from hypothesis import given, strategies as st
from hypothesis.errors import InvalidArgument


def always_passes(*args, **kwargs):
    """Stand-in for a fixed version of an inner test.

    For example, pytest-trio would take the inner test, wrap it in an
    async-to-sync converter, and use the new func (not always_passes).
    """


@given(st.integers())
def test_can_replace_inner_test(x):
    raise AssertionError("This should be replaced")


test_can_replace_inner_test.hypothesis.inner_test = always_passes


def decorator(func):
    """An example of a common decorator pattern."""

    @wraps(func)
    def inner(*args, **kwargs):
        return func(*args, **kwargs)

    return inner


@decorator
@given(st.integers())
def test_can_replace_when_decorated(x):
    raise AssertionError("This should be replaced")


test_can_replace_when_decorated.hypothesis.inner_test = always_passes


@pytest.mark.parametrize("x", [1, 2])
@given(y=st.integers())
def test_can_replace_when_parametrized(x, y):
    raise AssertionError("This should be replaced")


test_can_replace_when_parametrized.hypothesis.inner_test = always_passes


def test_can_replace_when_original_is_invalid():
    # Invalid: @given with too many positional arguments
    @given(st.integers(), st.integers())
    def invalid_test(x):
        raise AssertionError

    invalid_test.hypothesis.inner_test = always_passes

    # Even after replacing the inner test, calling the wrapper should still
    # fail.
    with pytest.raises(InvalidArgument, match="Too many positional arguments"):
        invalid_test()


def test_inner_is_original_even_when_invalid():
    def invalid_test(x):
        raise AssertionError

    original = invalid_test

    # Invalid: @given with no arguments
    invalid_test = given()(invalid_test)

    # Verify that the test is actually invalid
    with pytest.raises(
        InvalidArgument,
        match="given must be called with at least one argument",
    ):
        invalid_test()

    assert invalid_test.hypothesis.inner_test == original


def test_invokes_inner_function_with_args_by_name():
    # Regression test for https://github.com/HypothesisWorks/hypothesis/issues/3245
    @given(st.integers())
    def test(x):
        pass

    f = test.hypothesis.inner_test
    test.hypothesis.inner_test = wraps(f)(lambda **kw: f(**kw))
    test()