48 lines
1.3 KiB
Python
48 lines
1.3 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from functools import partial
|
||
|
|
from collections.abc import Callable
|
||
|
|
|
||
|
|
import pytest # type: ignore
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
AR = np.array(0)
|
||
|
|
AR.setflags(write=False)
|
||
|
|
|
||
|
|
KACF = frozenset({None, "K", "A", "C", "F"})
|
||
|
|
ACF = frozenset({None, "A", "C", "F"})
|
||
|
|
CF = frozenset({None, "C", "F"})
|
||
|
|
|
||
|
|
order_list: list[tuple[frozenset, Callable]] = [
|
||
|
|
(KACF, partial(np.ndarray, 1)),
|
||
|
|
(KACF, AR.tobytes),
|
||
|
|
(KACF, partial(AR.astype, int)),
|
||
|
|
(KACF, AR.copy),
|
||
|
|
(ACF, partial(AR.reshape, 1)),
|
||
|
|
(KACF, AR.flatten),
|
||
|
|
(KACF, AR.ravel),
|
||
|
|
(KACF, partial(np.array, 1)),
|
||
|
|
(CF, partial(np.zeros, 1)),
|
||
|
|
(CF, partial(np.ones, 1)),
|
||
|
|
(CF, partial(np.empty, 1)),
|
||
|
|
(CF, partial(np.full, 1, 1)),
|
||
|
|
(KACF, partial(np.zeros_like, AR)),
|
||
|
|
(KACF, partial(np.ones_like, AR)),
|
||
|
|
(KACF, partial(np.empty_like, AR)),
|
||
|
|
(KACF, partial(np.full_like, AR, 1)),
|
||
|
|
(KACF, partial(np.add, 1, 1)), # i.e. np.ufunc.__call__
|
||
|
|
(ACF, partial(np.reshape, AR, 1)),
|
||
|
|
(KACF, partial(np.ravel, AR)),
|
||
|
|
(KACF, partial(np.asarray, 1)),
|
||
|
|
(KACF, partial(np.asanyarray, 1)),
|
||
|
|
]
|
||
|
|
|
||
|
|
for order_set, func in order_list:
|
||
|
|
for order in order_set:
|
||
|
|
func(order=order)
|
||
|
|
|
||
|
|
invalid_orders = KACF - order_set
|
||
|
|
for order in invalid_orders:
|
||
|
|
with pytest.raises(ValueError):
|
||
|
|
func(order=order)
|