Move our next_*_normal helpers to internal.floats

honno · honno · commit d555dc139aec · 2021-12-02T14:43:03.000Z
diff --git a/hypothesis-python/src/hypothesis/internal/floats.py b/hypothesis-python/src/hypothesis/internal/floats.py
@@ -102,6 +102,17 @@ def next_down(value, width=64):
     return -next_up(-value, width)
 
 
+def next_down_normal(value, width, allow_subnormal):
+    value = next_down(value, width)
+    if (not allow_subnormal) and 0 < abs(value) < width_smallest_normals[width]:
+        return 0.0 if value > 0 else -width_smallest_normals[width]
+    return value
+
+
+def next_up_normal(value, width, allow_subnormal):
+    return -next_down_normal(-value, width, allow_subnormal)
+
+
 # Smallest positive non-zero numbers that is fully representable by an
 # IEEE-754 float, calculated with the width's associated minimum exponent.
 # Values from https://en.wikipedia.org/wiki/IEEE_754#Basic_and_interchange_formats
diff --git a/hypothesis-python/src/hypothesis/strategies/_internal/numbers.py b/hypothesis-python/src/hypothesis/strategies/_internal/numbers.py
@@ -31,8 +31,9 @@
     float_to_int,
     int_to_float,
     is_negative,
-    next_down,
+    next_down_normal,
     next_up,
+    next_up_normal,
     width_smallest_normals,
 )
 from hypothesis.internal.validation import (
@@ -279,17 +280,6 @@ def do_draw(self, data):
         return f
 
 
-def next_down_normal(value, width, allow_subnormal):
-    value = next_down(value, width)
-    if (not allow_subnormal) and 0 < abs(value) < width_smallest_normals[width]:
-        return 0.0 if value > 0 else -width_smallest_normals[width]
-    return value
-
-
-def next_up_normal(value, width, allow_subnormal):
-    return -next_down_normal(-value, width, allow_subnormal)
-
-
 @cacheable
 @defines_strategy(force_reusable_values=True)
 def floats(
