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
103
104
105
106
107
|
# 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
to_list = ak.operations.to_list
def test_UnmaskedArray():
content_float64 = ak.contents.NumpyArray(
np.array([0.25, 0.5, 3.5, 4.5, 5.5], dtype=np.float64)
)
array_float64 = ak.contents.UnmaskedArray(content_float64)
assert to_list(array_float64) == [0.25, 0.5, 3.5, 4.5, 5.5]
assert str(ak.operations.type(content_float64)) == "5 * float64"
assert str(ak.operations.type(ak.highlevel.Array(content_float64))) == "5 * float64"
assert str(ak.operations.type(array_float64)) == "5 * ?float64"
assert str(ak.operations.type(ak.highlevel.Array(array_float64))) == "5 * ?float64"
assert str(content_float64.form.type) == "float64"
assert str(array_float64.form.type) == "?float64"
assert np.can_cast(np.float32, np.float64) is True
assert np.can_cast(np.float64, np.float32, "unsafe") is True
assert np.can_cast(np.float64, np.int8, "unsafe") is True
content_float32 = ak.operations.values_astype(
content_float64, "float32", highlevel=False
)
array_float32 = ak.contents.UnmaskedArray(content_float32)
assert to_list(array_float32) == [0.25, 0.5, 3.5, 4.5, 5.5]
assert str(ak.operations.type(content_float32)) == "5 * float32"
assert str(ak.operations.type(ak.highlevel.Array(content_float32))) == "5 * float32"
assert str(ak.operations.type(array_float32)) == "5 * ?float32"
assert str(ak.operations.type(ak.highlevel.Array(array_float32))) == "5 * ?float32"
assert str(content_float32.form.type) == "float32"
assert str(array_float32.form.type) == "?float32"
content_int8 = ak.operations.values_astype(content_float64, "int8", highlevel=False)
array_int8 = ak.contents.UnmaskedArray(content_int8)
assert to_list(array_int8) == [0, 0, 3, 4, 5]
assert str(ak.operations.type(content_int8)) == "5 * int8"
assert str(ak.operations.type(ak.highlevel.Array(content_int8))) == "5 * int8"
assert str(ak.operations.type(array_int8)) == "5 * ?int8"
assert str(ak.operations.type(ak.highlevel.Array(array_int8))) == "5 * ?int8"
assert str(content_int8.form.type) == "int8"
assert str(array_int8.form.type) == "?int8"
content_from_int8 = ak.operations.values_astype(
content_int8, "float64", highlevel=False
)
array_from_int8 = ak.contents.UnmaskedArray(content_from_int8)
assert to_list(array_from_int8) == [0, 0, 3, 4, 5]
assert str(ak.operations.type(content_from_int8)) == "5 * float64"
assert (
str(ak.operations.type(ak.highlevel.Array(content_from_int8))) == "5 * float64"
)
assert str(ak.operations.type(array_from_int8)) == "5 * ?float64"
assert (
str(ak.operations.type(ak.highlevel.Array(array_from_int8))) == "5 * ?float64"
)
assert str(content_from_int8.form.type) == "float64"
assert str(array_from_int8.form.type) == "?float64"
def test_RegularArray_and_ListArray():
content = ak.contents.NumpyArray(
np.array([0.0, 1.1, 2.2, 3.3, 4.4, 5.5, 6.6, 7.7, 8.8, 9.9])
)
offsets = ak.index.Index64(np.array([0, 3, 3, 5, 6, 10, 10]))
listoffsetarray = ak.contents.ListOffsetArray(offsets, content)
regulararray = ak.contents.RegularArray(listoffsetarray, 2, zeros_length=0)
starts = ak.index.Index64(np.array([0, 1]))
stops = ak.index.Index64(np.array([2, 3]))
listarray = ak.contents.ListArray(starts, stops, regulararray)
assert str(ak.operations.type(content)) == "10 * float64"
assert str(ak.operations.type(regulararray)) == "3 * 2 * var * float64"
assert str(ak.operations.type(listarray)) == "2 * var * 2 * var * float64"
regulararray_int8 = ak.operations.values_astype(
regulararray, "int8", highlevel=False
)
assert str(ak.operations.type(regulararray_int8)) == "3 * 2 * var * int8"
listarray_bool = ak.operations.values_astype(listarray, "bool", highlevel=False)
assert str(ak.operations.type(listarray_bool)) == "2 * var * 2 * var * bool"
def test_ufunc_afterward():
assert (
ak.operations.values_astype(
ak.highlevel.Array([{"x": 1.1}, {"x": 3.3}]), np.float32
)["x"]
+ 1
).to_list() == [2.0999999046325684, 4.300000190734863]
def test_string():
assert to_list(
ak.operations.values_astype(
ak.highlevel.Array([{"x": 1.1, "y": "hello"}]), np.float32
)
) == [{"x": 1.100000023841858, "y": "hello"}]
|
