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
|
# BSD 3-Clause License; see https://github.com/scikit-hep/awkward/blob/main/LICENSE
from __future__ import annotations
import pytest # noqa: F401
import awkward as ak
to_list = ak.operations.to_list
def test():
array = ak.Array(
[[1.1, 2.2], [], [3.3, float("nan")], [float("-inf"), float("inf"), 7.7]]
)
assert to_list(
ak.operations.nan_to_num(
array, nan=999, posinf=float("-inf"), neginf=float("inf")
)
) == [[1.1, 2.2], [], [3.3, 999.0], [float("inf"), float("-inf"), 7.7]]
assert to_list(
ak.operations.nan_to_num(
array,
nan=[[-1, -2], [], [-3, -4], [-5, -6, -7]],
posinf=float("-inf"),
neginf=float("inf"),
)
) == [[1.1, 2.2], [], [3.3, -4], [float("inf"), float("-inf"), 7.7]]
assert to_list(
ak.operations.nan_to_num(
array,
nan=[[-1, -2], [], [-3, -4], [-5, -6, -7]],
posinf=[[1, 2], [], [3, 4], [5, 6, 7]],
neginf=float("inf"),
)
) == [[1.1, 2.2], [], [3.3, -4], [float("inf"), 6.0, 7.7]]
assert to_list(
ak.operations.nan_to_num(
array,
nan=[[-1, -2], [], [-3, -4], [-5, -6, -7]],
posinf=[[1, 2], [], [3, 4], [5, 6, 7]],
neginf=[[10, 20], [], [30, 40], [50, 60, 70]],
)
) == [[1.1, 2.2], [], [3.3, -4], [50.0, 6.0, 7.7]]
assert to_list(
ak.operations.nan_to_num(
array,
nan=[[-1, -2], [], [-3, -4], [-5, -6, -7]],
posinf=float("-inf"),
neginf=[[10, 20], [], [30, 40], [50, 60, 70]],
)
) == [[1.1, 2.2], [], [3.3, -4], [50.0, float("-inf"), 7.7]]
assert to_list(
ak.operations.nan_to_num(
array,
nan=999,
posinf=float("-inf"),
neginf=[[10, 20], [], [30, 40], [50, 60, 70]],
)
) == [[1.1, 2.2], [], [3.3, 999.0], [50.0, float("-inf"), 7.7]]
|
