def test_groupby():
array = ak.Array(
[
{"x": 1, "y": 1.1},
{"x": 2, "y": 2.2},
{"x": 1, "y": 1.1},
{"x": 3, "y": 3.3},
{"x": 1, "y": 1.1},
{"x": 2, "y": 2.2},
]
)
sorted = array[ak.argsort(array.x)]
assert sorted.x.tolist() == [1, 1, 1, 2, 2, 3]
assert ak.run_lengths(sorted.x).tolist() == [3, 2, 1]
assert ak.unflatten(sorted, ak.run_lengths(sorted.x)).tolist() == [
[{"x": 1, "y": 1.1}, {"x": 1, "y": 1.1}, {"x": 1, "y": 1.1}],
[{"x": 2, "y": 2.2}, {"x": 2, "y": 2.2}],
[{"x": 3, "y": 3.3}],
]
array = ak.Array(
[
[{"x": 1, "y": 1.1}, {"x": 2, "y": 2.2}, {"x": 1, "y": 1.1}],
[{"x": 3, "y": 3.3}, {"x": 1, "y": 1.1}, {"x": 2, "y": 2.2}],
]
)
sorted = array[ak.argsort(array.x)]
assert sorted.x.tolist() == [[1, 1, 2], [1, 2, 3]]
assert ak.run_lengths(sorted.x).tolist() == [[2, 1], [1, 1, 1]]
counts = ak.flatten(ak.run_lengths(sorted.x), axis=None)
assert ak.unflatten(sorted, counts, axis=-1).tolist() == [
[[{"x": 1, "y": 1.1}, {"x": 1, "y": 1.1}], [{"x": 2, "y": 2.2}]],
[[{"x": 1, "y": 1.1}], [{"x": 2, "y": 2.2}], [{"x": 3, "y": 3.3}]],
]
def test():
array = ak.Array([3, 3, 3, 5, 5, 9, 9, 9, 9, 1, 3, 3])
assert ak.run_lengths(array).tolist() == [3, 2, 4, 1, 2]
array = ak.Array([[3, 3, 3, 5], [5], [], [9, 9], [9, 9], [1, 3, 3]])
assert ak.run_lengths(array).tolist() == [[3, 1], [1], [], [2], [2], [1, 2]]
array = ak.repartition(ak.Array([3, 3, 3, 5, 5, 9, 9, 9, 9, 1, 3, 3]), 7)
assert ak.run_lengths(array).tolist() == [3, 2, 4, 1, 2]
array = ak.repartition(
ak.Array([[3, 3, 3, 5], [5], [], [9, 9], [9, 9], [1, 3, 3]]), 4
)
assert ak.run_lengths(array).tolist() == [[3, 1], [1], [], [2], [2], [1, 2]]
def test_onstrings1():
data = ak.Array(["one", "one", "one", "two", "two", "three", "two", "two"])
assert ak.run_lengths(data).tolist() == [3, 2, 1, 2]
data = ak._v2.Array(
["one", "one", "one", "two", "two", "three", "two", "two"])
assert ak._v2.operations.structure.run_lengths(data).tolist() == [
3, 2, 1, 2
]
def test():
layout = ak.layout.IndexedArray64(
ak.layout.Index64(np.array([3, 1, 0, 2])),
ak.layout.ListOffsetArray64(
ak.layout.Index64(np.array([0, 3, 6, 9, 12])),
ak.layout.NumpyArray(np.array([0, 0, 0, 1, 1, 1, 2, 2, 3, 3, 3, 3])),
),
)
assert ak.unflatten(
layout, ak.flatten(ak.run_lengths(layout)), axis=1
).tolist() == [[[3, 3, 3]], [[1, 1, 1]], [[0, 0, 0]], [[2, 2], [3]]]
def test_onstrings2():
data = ak.Array([["one", "one"], ["one", "two", "two"],
["three", "two", "two"]])
assert ak.run_lengths(data).tolist() == [[2], [1, 2], [1, 2]]
def test_onstrings1():
data = ak.Array(["one", "one", "one", "two", "two", "three", "two", "two"])
assert ak.run_lengths(data).tolist() == [3, 2, 1, 2]