1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
# BSD 3-Clause License; see https://github.com/scikit-hep/awkward/blob/main/LICENSE
from __future__ import annotations
import numpy as np # noqa: F401
import pytest # noqa: F401
import awkward as ak
to_list = ak.operations.to_list
def test_sort():
data = ak.Array([[7, 5, 7], [], [2], [8, 2]])
assert to_list(ak.operations.sort(data)) == [
[5, 7, 7],
[],
[2],
[2, 8],
]
def test_argsort():
data = ak.Array([[7, 5, 7], [], [2], [8, 2]])
index = ak.operations.argsort(data)
assert to_list(data[index]) == [[5, 7, 7], [], [2], [2, 8]]
|
