literal.py 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445
  1. from functools import partial
  2. from typing import Callable, List, Tuple
  3. import pytest # type: ignore
  4. import numpy as np
  5. AR = np.array(0)
  6. AR.setflags(write=False)
  7. KACF = frozenset({None, "K", "A", "C", "F"})
  8. ACF = frozenset({None, "A", "C", "F"})
  9. CF = frozenset({None, "C", "F"})
  10. order_list: List[Tuple[frozenset, Callable]] = [
  11. (KACF, partial(np.ndarray, 1)),
  12. (KACF, AR.tobytes),
  13. (KACF, partial(AR.astype, int)),
  14. (KACF, AR.copy),
  15. (ACF, partial(AR.reshape, 1)),
  16. (KACF, AR.flatten),
  17. (KACF, AR.ravel),
  18. (KACF, partial(np.array, 1)),
  19. (CF, partial(np.zeros, 1)),
  20. (CF, partial(np.ones, 1)),
  21. (CF, partial(np.empty, 1)),
  22. (CF, partial(np.full, 1, 1)),
  23. (KACF, partial(np.zeros_like, AR)),
  24. (KACF, partial(np.ones_like, AR)),
  25. (KACF, partial(np.empty_like, AR)),
  26. (KACF, partial(np.full_like, AR, 1)),
  27. (KACF, partial(np.add, 1, 1)), # i.e. np.ufunc.__call__
  28. (ACF, partial(np.reshape, AR, 1)),
  29. (KACF, partial(np.ravel, AR)),
  30. (KACF, partial(np.asarray, 1)),
  31. (KACF, partial(np.asanyarray, 1)),
  32. ]
  33. for order_set, func in order_list:
  34. for order in order_set:
  35. func(order=order)
  36. invalid_orders = KACF - order_set
  37. for order in invalid_orders:
  38. with pytest.raises(ValueError):
  39. func(order=order)