From 949bda505d226150bae0d7324ca675899c8396b9 Mon Sep 17 00:00:00 2001 From: Tobias Pitters Date: Wed, 27 May 2020 23:57:30 +0200 Subject: refactor quantile tests --- numpy/lib/tests/test_function_base.py | 6 ++---- 1 file changed, 2 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 b9ee49f16..3c0c8ca89 100644 --- a/numpy/lib/tests/test_function_base.py +++ b/numpy/lib/tests/test_function_base.py @@ -3113,16 +3113,14 @@ class TestQuantile: # test that the return value of quantile is monotonic if p0 is ordered p0 = np.arange(0, 1, 0.01) quantile = np.quantile(np.array([0, 1, 1, 2, 2, 3, 3, 4, 5, 5, 1, 1, 9, 9, 9, 8, 8, 7]) * 0.1, p0) - equals_sorted = np.sort(quantile) == quantile - assert equals_sorted.all() + assert_equal(np.sort(quantile), quantile) @hypothesis.given(arr=arrays(dtype=np.float, shape=st.integers(min_value=3, max_value=1000), elements=st.floats(allow_infinity=False, allow_nan=False))) def test_quantile_monotonic_hypo(self, arr): p0 = np.arange(0, 1, 0.01) quantile = np.quantile(arr, p0) - equals_sorted = np.sort(quantile) == quantile - assert equals_sorted.all() + assert_equal(np.sort(quantile), quantile) class TestLerp: -- cgit v1.2.1