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
|
# BSD 3-Clause License; see https://github.com/scikit-hep/awkward/blob/main/LICENSE
from __future__ import annotations
import pytest
import awkward as ak
def test():
array = ak.Array(
["is", "this", "real", "life", "or", "is", "this", "just", "fantasy"]
)
with pytest.raises(ValueError, match=r"strings cannot be directly flattened"):
ak.flatten(array, axis=1)
array = ak.Array(
["is", "this", "real", "life", "or", "is", "this", "just", "fantasy"]
)
assert ak.to_layout(array) == ak.flatten(array, axis=0, highlevel=False)
array_ragged = ak.Array(
[["is", "this", "real", "life"], ["or", "is", "this", "just", "fantasy"]]
)
assert ak.almost_equal(array, ak.flatten(array_ragged, axis=1))
with pytest.raises(ValueError, match=r"strings cannot be directly flattened"):
ak.flatten(array_ragged, axis=2)
|
