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
|
# 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_example():
x = ak.operations.from_iter([[1, 2, 3, None], [], [4, 5]])
y = ak.operations.from_iter([100, 200, 300])
ttx = ak.highlevel.Array(x.layout.to_typetracer())
tty = ak.highlevel.Array(y.layout.to_typetracer())
assert (x + y).layout.form == (ttx + tty).layout.form
assert (x + np.sin(y)).layout.form == (ttx + np.sin(tty)).layout.form
x = ak.highlevel.Array(
ak.contents.ListArray(x.layout.starts, x.layout.stops, x.layout.content)
)
ttx = ak.highlevel.Array(x.layout.to_typetracer())
assert (x + x).layout.form == (ttx + ttx).layout.form
|
