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
30
31
32
33
34
35
36
37
38
39
40
| """ common utilities """
| from __future__ import annotations
|
| from typing import (
| Any,
| Literal,
| )
|
|
| def _mklbl(prefix: str, n: int):
| return [f"{prefix}{i}" for i in range(n)]
|
|
| def check_indexing_smoketest_or_raises(
| obj,
| method: Literal["iloc", "loc"],
| key: Any,
| axes: Literal[0, 1] | None = None,
| fails=None,
| ) -> None:
| if axes is None:
| axes_list = [0, 1]
| else:
| assert axes in [0, 1]
| axes_list = [axes]
|
| for ax in axes_list:
| if ax < obj.ndim:
| # create a tuple accessor
| new_axes = [slice(None)] * obj.ndim
| new_axes[ax] = key
| axified = tuple(new_axes)
| try:
| getattr(obj, method).__getitem__(axified)
| except (IndexError, TypeError, KeyError) as detail:
| # if we are in fails, the ok, otherwise raise it
| if fails is not None:
| if isinstance(detail, fails):
| return
| raise
|
|
