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
|
# BSD 3-Clause License; see https://github.com/scikit-hep/awkward/blob/main/LICENSE
from __future__ import annotations
import numpy as np
import pytest # noqa: F401
import awkward as ak
def test():
a = ak.operations.values_astype(ak.Array([1, None]), np.float32)
assert ak.operations.fill_none(a, np.float32(0)).tolist() == [1, 0]
assert str(ak.operations.fill_none(a, np.float32(0)).type) == "2 * float32"
assert ak.operations.fill_none(a, np.array(0, np.float32)).tolist() == [1, 0]
assert (
str(ak.operations.fill_none(a, np.array(0, np.float32)).type) == "2 * float32"
)
assert ak.operations.fill_none(a, np.array([0], np.float32)).tolist() == [
1,
[0],
]
assert (
str(ak.operations.fill_none(a, np.array([0], np.float32)).type)
== "2 * union[float32, 1 * float32]"
)
assert ak.operations.fill_none(a, np.array([[0]], np.float32)).tolist() == [
1,
[[0]],
]
assert (
str(ak.operations.fill_none(a, np.array([[0]], np.float32)).type)
== "2 * union[float32, 1 * 1 * float32]"
)
assert ak.operations.fill_none(a, 0).to_list() == [1, 0]
assert str(ak.operations.fill_none(a, 0).type) == "2 * float64"
assert ak.operations.fill_none(a, [0]).to_list() == [1, [0]]
assert str(ak.operations.fill_none(a, [0]).type) == "2 * union[float32, 1 * int64]"
assert ak.operations.fill_none(a, [[0]]).to_list() == [1, [[0]]]
assert (
str(ak.operations.fill_none(a, [[0]]).type)
== "2 * union[float32, 1 * var * int64]"
)
|
