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
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
# coding=utf-8
#
# This file is part of Hypothesis, which may be found at
# https://github.com/HypothesisWorks/hypothesis-python
#
# Most of this work is copyright (C) 2013-2015 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 http://mozilla.org/MPL/2.0/.
#
# END HEADER
from __future__ import division, print_function, absolute_import
import pytest
from hypothesis import strategies as st
from hypothesis import given
from hypothesis.errors import Frozen
from hypothesis.internal.conjecture.data import Status, StopTest, TestData
from hypothesis.searchstrategy.strategies import SearchStrategy
def bogus_dist(dist, n):
assert False
@given(st.binary())
def test_buffer_draws_as_self(buf):
x = TestData.for_buffer(buf)
assert x.draw_bytes(len(buf), bogus_dist) == buf
def test_cannot_draw_after_freeze():
x = TestData.for_buffer(b'hi')
x.draw_bytes(1)
x.freeze()
with pytest.raises(Frozen):
x.draw_bytes(1)
def test_can_double_freeze():
x = TestData.for_buffer(b'hi')
x.freeze()
assert x.frozen
x.freeze()
assert x.frozen
def test_can_draw_zero_bytes():
x = TestData.for_buffer(b'')
for _ in range(10):
assert x.draw_bytes(0) == b''
def test_draw_past_end_sets_overflow():
x = TestData.for_buffer(bytes(5))
with pytest.raises(StopTest) as e:
x.draw_bytes(6)
assert e.value.testcounter == x.testcounter
assert x.frozen
assert x.status == Status.OVERRUN
def test_notes_repr():
x = TestData.for_buffer(b'')
x.note(b'hi')
assert repr(b'hi') in x.output
def test_can_mark_interesting():
x = TestData.for_buffer(bytes())
with pytest.raises(StopTest):
x.mark_interesting()
assert x.frozen
assert x.status == Status.INTERESTING
def test_can_mark_invalid():
x = TestData.for_buffer(bytes())
with pytest.raises(StopTest):
x.mark_invalid()
assert x.frozen
assert x.status == Status.INVALID
class BoomStrategy(SearchStrategy):
def do_draw(self, data):
data.draw_bytes(1)
raise ValueError()
def test_closes_interval_on_error_in_strategy():
x = TestData.for_buffer(b'hi')
with pytest.raises(ValueError):
x.draw(BoomStrategy())
x.freeze()
assert len(x.intervals) == 1
class BigStrategy(SearchStrategy):
def do_draw(self, data):
data.draw_bytes(10 ** 6)
def test_does_not_double_freeze_in_interval_close():
x = TestData.for_buffer(b'hi')
with pytest.raises(StopTest):
x.draw(BigStrategy())
assert x.frozen
assert len(x.intervals) == 0
|
