Skip to content

Commit 41b15e3

Browse files
committed
don't permit >128 bit integers
1 parent 6a981fc commit 41b15e3

File tree

2 files changed

+25
-5
lines changed

2 files changed

+25
-5
lines changed

hypothesis-python/src/hypothesis/internal/conjecture/data.py

+10-2
Original file line numberDiff line numberDiff line change
@@ -1072,9 +1072,17 @@ def __repr__(self):
10721072

10731073
def ir_value_permitted(value, ir_type, kwargs):
10741074
if ir_type == "integer":
1075-
if kwargs["min_value"] is not None and value < kwargs["min_value"]:
1075+
min_value = kwargs["min_value"]
1076+
max_value = kwargs["max_value"]
1077+
shrink_towards = kwargs["shrink_towards"]
1078+
if min_value is not None and value < min_value:
10761079
return False
1077-
if kwargs["max_value"] is not None and value > kwargs["max_value"]:
1080+
if max_value is not None and value > max_value:
1081+
return False
1082+
1083+
if (max_value is None or min_value is None) and (
1084+
value - shrink_towards
1085+
).bit_length() >= 128:
10781086
return False
10791087

10801088
return True

hypothesis-python/tests/conjecture/test_ir.py

+15-3
Original file line numberDiff line numberDiff line change
@@ -523,9 +523,21 @@ def test_all_children_are_permitted_values(ir_type_and_kwargs):
523523
@pytest.mark.parametrize(
524524
"value, ir_type, kwargs, permitted",
525525
[
526-
(0, "integer", {"min_value": 1, "max_value": 2}, False),
527-
(2, "integer", {"min_value": 0, "max_value": 1}, False),
528-
(10, "integer", {"min_value": 0, "max_value": 20}, True),
526+
(0, "integer", {"min_value": 1, "max_value": 2, "shrink_towards": 0}, False),
527+
(2, "integer", {"min_value": 0, "max_value": 1, "shrink_towards": 0}, False),
528+
(10, "integer", {"min_value": 0, "max_value": 20, "shrink_towards": 0}, True),
529+
(
530+
int(2**128 / 2) - 1,
531+
"integer",
532+
{"min_value": None, "max_value": None, "shrink_towards": 0},
533+
True,
534+
),
535+
(
536+
int(2**128 / 2),
537+
"integer",
538+
{"min_value": None, "max_value": None, "shrink_towards": 0},
539+
False,
540+
),
529541
(
530542
math.nan,
531543
"float",

0 commit comments

Comments
 (0)