Method assert_partitioned

numpy/_core/tests/test_multiarray.py:3588–3595

(self, d, kth)

                    assert_equal(dc, np.partition(d1, i, axis=1, kind=k))

    def assert_partitioned(self, d, kth):
        prev = 0
        for k in np.sort(kth):
            assert_array_compare(operator.__le__, d[prev:k], d[k],
                    err_msg=f'kth {k}')
            assert_((d[k:] >= d[k]).all(),
                    msg=f"kth {k}, {d[k:]!r} not greater equal {d[k]!r}")
            prev = k + 1

    def test_partition_iterative(self):
        d = np.arange(17)