1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
# BSD 3-Clause License; see https://github.com/scikit-hep/awkward/blob/main/LICENSE
from __future__ import annotations
import numpy as np # noqa: F401
import pytest
import awkward as ak
def test_where():
result = ak.where(
[True, False, False], ak.to_backend([1, 2, 3], "typetracer"), [4, 5, 6]
)
assert ak.backend(result) == "typetracer"
def test_almost_equal():
with pytest.raises(NotImplementedError):
ak.almost_equal(
[True, False, False], ak.to_backend([True, False, False], "typetracer")
)
|
