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
|
# BSD 3-Clause License; see https://github.com/scikit-hep/awkward/blob/main/LICENSE
from __future__ import annotations
import numpy as np
import pytest
import awkward as ak
def test_binary():
ak_array = ak.Array(np.arange(10, dtype="<u4"))
np_array = np.arange(10, dtype=">u4")
with pytest.raises(TypeError):
# ak.array_equal now overrides np.array_equal, and requires
# both arrays to be valid within awkward.
assert np.array_equal(ak_array, np_array)
def test_different_backend():
pytest.importorskip("jax")
ak.jax.register_and_check()
left = ak.Array([0, 1, 2], backend="jax")
right = ak.Array([0, 1, 2], backend="cpu")
with pytest.raises(
ValueError, match="cannot operate on arrays with incompatible backends"
):
assert np.array_equal(left, right)
|
