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
|
# BSD 3-Clause License; see https://github.com/scikit-hep/awkward/blob/main/LICENSE
from __future__ import annotations
import numpy as np
import pytest
import awkward as ak
def test():
assert ak.operations.is_none(ak.Array([1, 2, 3, None, 5])).to_list() == [
False,
False,
False,
True,
False,
]
assert ak.operations.is_none(ak.Array([[1, 2, 3], [], [None, 5]])).to_list() == [
False,
False,
False,
]
assert ak.operations.is_none(
ak.Array([[1, 2, 3], [], [None, 5]]), axis=1
).to_list() == [
[False, False, False],
[],
[True, False],
]
assert ak.operations.is_none(
ak.Array([[1, None, 2, 3], [], [None, 5]]), axis=1
).to_list() == [
[False, True, False, False],
[],
[True, False],
]
assert ak.operations.is_none(
ak.Array([[1, None, 2, 3], [], [None, 5]]), axis=-1
).to_list() == [
[False, True, False, False],
[],
[True, False],
]
assert ak.operations.is_none(
ak.Array([[1, None, 2, 3], [], [None, 5]]), axis=-2
).to_list() == [
False,
False,
False,
]
assert ak.operations.is_none(
ak.Array([[1, None, 2, 3], None, [None, 5]]), axis=-2
).to_list() == [False, True, False]
with pytest.raises(ValueError):
ak.operations.is_none(ak.Array([[1, None, 2, 3], [], [None, 5]]), axis=-3)
one = ak.operations.from_iter([1, None, 3], highlevel=False)
two = ak.operations.from_iter([[], [1], None, [3, 3, 3]], highlevel=False)
tags = ak.index.Index8(np.array([0, 1, 1, 0, 0, 1, 1], dtype=np.int8))
index = ak.index.Index64(np.array([0, 0, 1, 1, 2, 2, 3], dtype=np.int64))
array = ak.Array(ak.contents.UnionArray(tags, index, [one, two]), check_valid=True)
assert ak.to_list(array) == [1, [], [1], None, 3, None, [3, 3, 3]]
assert ak.to_list(ak.operations.is_none(array)) == [
False,
False,
False,
True,
False,
True,
False,
]
|
