File: test_3619_fancy_indexing_with_awkward_arrays.py

package info (click to toggle)
python-awkward 2.8.9-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 24,932 kB
  • sloc: python: 178,875; cpp: 33,828; sh: 432; makefile: 21; javascript: 8
file content (21 lines) | stat: -rw-r--r-- 657 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
# BSD 3-Clause License; see https://github.com/scikit-hep/awkward/blob/main/LICENSE
#
from __future__ import annotations

import pytest

import awkward as ak


def test():
    # taking the array directly from the issue: https://github.com/scikit-hep/awkward/issues/3619
    a = ak.Array([[1, 5], [3], [4]])

    # this is a valid case, where the index is not out of bounds
    c_valid = ak.Array([[1, 1], [0], [0]])
    assert a[c_valid].tolist() == [[5, 5], [3], [4]]

    # this is an invalid case, where the index for the second element `[1]` is out of bounds
    c_oob = ak.Array([[1, 1], [1], [0]])
    with pytest.raises(IndexError):
        a[c_oob]