-
Notifications
You must be signed in to change notification settings - Fork 575
/
test_lookup_py39.py
95 lines (70 loc) · 2.82 KB
/
test_lookup_py39.py
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
# This file is part of Hypothesis, which may be found at
# https://github.com/HypothesisWorks/hypothesis/
#
# Most of this work is copyright (C) 2013-2021 David R. MacIver
# (david@drmaciver.com), but it contains contributions by others. See
# CONTRIBUTING.rst for a full list of people who may hold copyright, and
# consult the git log if you need to determine who owns an individual
# contribution.
#
# 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/.
#
# END HEADER
import dataclasses
import typing
import pytest
from hypothesis import given, strategies as st
from hypothesis.errors import InvalidArgument
from tests.common.debug import find_any
@pytest.mark.parametrize(
"annotated_type,expected_strategy_repr",
[
(typing.Annotated[int, "foo"], "integers()"),
(typing.Annotated[typing.List[float], "foo"], "lists(floats())"),
(typing.Annotated[typing.Annotated[str, "foo"], "bar"], "text()"),
(
typing.Annotated[
typing.Annotated[typing.List[typing.Dict[str, bool]], "foo"], "bar"
],
"lists(dictionaries(keys=text(), values=booleans()))",
),
],
)
def test_typing_Annotated(annotated_type, expected_strategy_repr):
assert repr(st.from_type(annotated_type)) == expected_strategy_repr
PositiveInt = typing.Annotated[int, st.integers(min_value=1)]
MoreThenTenInt = typing.Annotated[PositiveInt, st.integers(min_value=10 + 1)]
WithTwoStrategies = typing.Annotated[int, st.integers(), st.none()]
ExtraAnnotationNoStrategy = typing.Annotated[PositiveInt, "metadata"]
def arg_positive(x: PositiveInt):
assert x > 0
def arg_more_than_ten(x: MoreThenTenInt):
assert x > 10
@given(st.data())
def test_annotated_positive_int(data):
data.draw(st.builds(arg_positive))
@given(st.data())
def test_annotated_more_than_ten(data):
data.draw(st.builds(arg_more_than_ten))
@given(st.data())
def test_annotated_with_two_strategies(data):
assert data.draw(st.from_type(WithTwoStrategies)) is None
@given(st.data())
def test_annotated_extra_metadata(data):
assert data.draw(st.from_type(ExtraAnnotationNoStrategy)) > 0
@dataclasses.dataclass
class User:
id: int
following: list["User"] # works with typing.List
def test_string_forward_ref_message():
# See https://github.com/HypothesisWorks/hypothesis/issues/3016
s = st.builds(User)
with pytest.raises(InvalidArgument, match="`from __future__ import annotations`"):
s.example()
def test_issue_3080():
# Check for https://github.com/HypothesisWorks/hypothesis/issues/3080
s = st.from_type(typing.Union[list[int], int])
find_any(s, lambda x: isinstance(x, int))
find_any(s, lambda x: isinstance(x, list))