From 214e8302ff92c002853ab03427cae2448a812f7c Mon Sep 17 00:00:00 2001 From: Tobias Pitters Date: Wed, 27 May 2020 19:50:49 +0200 Subject: fix lerp function and corresponding tests --- numpy/lib/tests/test_function_base.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) (limited to 'numpy/lib/tests/test_function_base.py') diff --git a/numpy/lib/tests/test_function_base.py b/numpy/lib/tests/test_function_base.py index 96e3135ef..5468901b4 100644 --- a/numpy/lib/tests/test_function_base.py +++ b/numpy/lib/tests/test_function_base.py @@ -3140,8 +3140,10 @@ class TestLerp: @hypothesis.given(t=st.floats(allow_nan=False, allow_infinity=False, min_value=0, max_value=1), - a=st.floats(allow_nan=False, allow_infinity=False), - b=st.floats(allow_nan=False, allow_infinity=False)) + a=st.floats(allow_nan=False, allow_infinity=False, + width=32), + b=st.floats(allow_nan=False, allow_infinity=False, + width=32)) def test_lerp_bounded(self, t, a, b): if a <= b: assert a <= np.lib.function_base._lerp(a, b, t) <= b @@ -3150,8 +3152,10 @@ class TestLerp: @hypothesis.given(t=st.floats(allow_nan=False, allow_infinity=False, min_value=0, max_value=1), - a=st.floats(allow_nan=False, allow_infinity=False), - b=st.floats(allow_nan=False, allow_infinity=False)) + a=st.floats(allow_nan=False, allow_infinity=False, + width=32), + b=st.floats(allow_nan=False, allow_infinity=False, + width=32)) def test_lerp_symmetric(self, t, a, b): # double subtraction is needed to remove the extra precision that t < 0.5 has assert np.lib.function_base._lerp(a, b, 1 - (1 - t)) == np.lib.function_base._lerp(b, a, 1 - t) -- cgit v1.2.1