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():
x = ak.contents.IndexedArray(
ak.index.Index64(np.array([0, 0, 1], dtype=np.int64)),
ak.contents.NumpyArray(np.array([9, 6, 5], dtype=np.int16)),
parameters={"money": "doesn't buy happiness"},
)
y = ak.contents.IndexedArray(
ak.index.Index64(np.array([0, 1, 2, 4, 3], dtype=np.int64)),
ak.contents.NumpyArray(np.array([9, 6, 5, 8, 2], dtype=np.int16)),
parameters={"age": "number"},
)
# Test that we invoke the merge pathway
z = x._reverse_merge(y)
assert z.to_list() == [9, 6, 5, 2, 8, 9, 9, 6]
assert z.parameters == {"money": "doesn't buy happiness", "age": "number"}
|