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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
# 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
def test_but_first_fix_sort():
assert ak.operations.is_valid(
ak.operations.sort(ak.Array(["one", "two", "three"]), axis=-1)
)
def test_argsort():
array = ak.Array(["one", "two", "three", "four", "five", "six", "seven", "eight"])
assert ak.operations.argsort(array, axis=-1).to_list() == [
7,
4,
3,
0,
6,
5,
2,
1,
]
array = ak.Array(
[["twotwo", "two", "three"], ["four", "five"], [], ["six", "seven", "eight"]]
)
assert ak.operations.argsort(array, axis=-1).to_list() == [
[2, 1, 0],
[1, 0],
[],
[2, 1, 0],
]
array = ak.Array(
[
[["twotwo", "two"], ["three"]],
[["four", "five"]],
[],
[["six"], ["seven", "eight"]],
]
)
assert ak.operations.argsort(array, axis=-1).to_list() == [
[[1, 0], [0]],
[[1, 0]],
[],
[[0], [1, 0]],
]
def test_sort():
array = ak.Array(["one", "two", "three", "four", "five", "six", "seven", "eight"])
assert ak.operations.sort(array, axis=-1).to_list() == [
"eight",
"five",
"four",
"one",
"seven",
"six",
"three",
"two",
]
array = ak.Array(
[["twotwo", "two", "three"], ["four", "five"], [], ["six", "seven", "eight"]]
)
assert ak.operations.sort(array, axis=-1).to_list() == [
["three", "two", "twotwo"],
["five", "four"],
[],
["eight", "seven", "six"],
]
array = ak.Array(
[
[["twotwo", "two"], ["three"]],
[["four", "five"]],
[],
[["six"], ["seven", "eight"]],
]
)
assert ak.operations.sort(array, axis=-1).to_list() == [
[["two", "twotwo"], ["three"]],
[["five", "four"]],
[],
[["six"], ["eight", "seven"]],
]
|
