Na}{

File Content: `test_indexing_functions.py`

import pytest

from numpy import array_api as xp


@pytest.mark.parametrize(
    "x, indices, axis, expected",
    [
        ([2, 3], [1, 1, 0], 0,  [3, 3, 2]),
        ([2, 3], [1, 1, 0], -1, [3, 3, 2]),
        ([[2, 3]], [1], -1, [[3]]),
        ([[2, 3]], [0, 0], 0, [[2, 3], [2, 3]]),
    ],
)
def test_take_function(x, indices, axis, expected):
    """
    Indices respect relative order of a descending stable-sort

    See https://github.com/numpy/numpy/issues/20778
    """
    x = xp.asarray(x)
    indices = xp.asarray(indices)
    out = xp.take(x, indices, axis=axis)
    assert xp.all(out == xp.asarray(expected))

Name	Size	Permission
__pycache__	---	0755
__init__.py	282 bytes	0644
test_array_object.py	17035 bytes	0644
test_creation_functions.py	5023 bytes	0644
test_data_type_functions.py	1018 bytes	0644
test_elementwise_functions.py	3800 bytes	0644
test_indexing_functions.py	627 bytes	0644
test_manipulation_functions.py	1087 bytes	0644
test_set_functions.py	546 bytes	0644
test_sorting_functions.py	602 bytes	0644
test_validation.py	676 bytes	0644

Name

Size

Permission

Action

__pycache__

---

0755