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
93
94
95
96
97
98
99
100
101
102
|
# 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
import awkward as ak
to_list = ak.operations.to_list
empty = ak.highlevel.Array(
ak.contents.RegularArray(
ak.highlevel.Array([[1, 2, 3], [], [4, 5]]).layout, 0, zeros_length=0
)
)
def test_ListOffsetArray_rpad_and_clip():
array = ak.highlevel.Array([[1, 2, 3], [], [4, 5]])
assert ak.operations.pad_none(array, 0, clip=True).to_list() == [
[],
[],
[],
]
array = ak.highlevel.Array([[1, 2, 3], [], [4, 5]])
assert ak.operations.pad_none(array, 0).to_list() == [
[1, 2, 3],
[],
[4, 5],
]
def test_to_ListOffsetArray64():
assert ak.operations.from_regular(empty).to_list() == []
def test_carry():
assert empty[[]].to_list() == []
def test_num():
assert ak.operations.num(empty, axis=0) == 0
assert ak.operations.num(empty, axis=1).to_list() == []
assert ak.operations.num(empty, axis=2).to_list() == []
def test_flatten():
assert ak.operations.flatten(empty, axis=0).to_list() == []
assert ak.operations.flatten(empty, axis=1).to_list() == []
assert ak.operations.flatten(empty, axis=2).to_list() == []
def test_mergeable():
assert ak.operations.concatenate([empty, empty]).to_list() == []
def test_fillna():
assert ak.operations.fill_none(empty, 5, axis=0).to_list() == []
def test_pad_none():
assert ak.operations.pad_none(empty, 0, axis=0).to_list() == []
assert ak.operations.pad_none(empty, 0, axis=1).to_list() == []
assert ak.operations.pad_none(empty, 0, axis=2).to_list() == []
assert ak.operations.pad_none(empty, 1, axis=0).to_list() == [None]
assert ak.operations.pad_none(empty, 1, axis=1).to_list() == []
assert ak.operations.pad_none(empty, 1, axis=2).to_list() == []
assert ak.operations.pad_none(empty, 0, axis=0, clip=True).to_list() == []
assert ak.operations.pad_none(empty, 0, axis=1, clip=True).to_list() == []
assert ak.operations.pad_none(empty, 0, axis=2, clip=True).to_list() == []
assert ak.operations.pad_none(empty, 1, axis=0, clip=True).to_list() == [None]
assert ak.operations.pad_none(empty, 1, axis=1, clip=True).to_list() == []
assert ak.operations.pad_none(empty, 1, axis=2, clip=True).to_list() == []
def test_reduce():
assert ak.operations.sum(empty, axis=0).to_list() == []
def test_localindex():
assert ak.operations.local_index(empty, axis=0).to_list() == []
assert ak.operations.local_index(empty, axis=1).to_list() == []
assert ak.operations.local_index(empty, axis=2).to_list() == []
def test_combinations():
assert ak.operations.combinations(empty, 2, axis=0).to_list() == []
assert ak.operations.combinations(empty, 2, axis=1).to_list() == []
assert ak.operations.combinations(empty, 2, axis=2).to_list() == []
def test_getitem():
with pytest.raises(IndexError):
empty[0,]
jagged = ak.highlevel.Array([[]])[0:0]
assert empty[jagged].to_list() == []
|
