41 lines
1021 B
Python
41 lines
1021 B
Python
|
""" 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
|