Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.9/dist-packages/scipy/_lib/array_api_compat/numpy/_aliases.py: 97%

1from __future__ import annotations 

2 

3from functools import partial 

4 

5from ..common import _aliases 

6 

7from .._internal import get_xp 

8 

9asarray = asarray_numpy = partial(_aliases._asarray, namespace='numpy') 

10asarray.__doc__ = _aliases._asarray.__doc__ 

11del partial 

12 

13import numpy as np 

14bool = np.bool_ 

15 

16# Basic renames 

17acos = np.arccos 

18acosh = np.arccosh 

19asin = np.arcsin 

20asinh = np.arcsinh 

21atan = np.arctan 

22atan2 = np.arctan2 

23atanh = np.arctanh 

24bitwise_left_shift = np.left_shift 

25bitwise_invert = np.invert 

26bitwise_right_shift = np.right_shift 

27concat = np.concatenate 

28pow = np.power 

29 

30arange = get_xp(np)(_aliases.arange) 

31empty = get_xp(np)(_aliases.empty) 

32empty_like = get_xp(np)(_aliases.empty_like) 

33eye = get_xp(np)(_aliases.eye) 

34full = get_xp(np)(_aliases.full) 

35full_like = get_xp(np)(_aliases.full_like) 

36linspace = get_xp(np)(_aliases.linspace) 

37ones = get_xp(np)(_aliases.ones) 

38ones_like = get_xp(np)(_aliases.ones_like) 

39zeros = get_xp(np)(_aliases.zeros) 

40zeros_like = get_xp(np)(_aliases.zeros_like) 

41UniqueAllResult = get_xp(np)(_aliases.UniqueAllResult) 

42UniqueCountsResult = get_xp(np)(_aliases.UniqueCountsResult) 

43UniqueInverseResult = get_xp(np)(_aliases.UniqueInverseResult) 

44unique_all = get_xp(np)(_aliases.unique_all) 

45unique_counts = get_xp(np)(_aliases.unique_counts) 

46unique_inverse = get_xp(np)(_aliases.unique_inverse) 

47unique_values = get_xp(np)(_aliases.unique_values) 

48astype = _aliases.astype 

49std = get_xp(np)(_aliases.std) 

50var = get_xp(np)(_aliases.var) 

51permute_dims = get_xp(np)(_aliases.permute_dims) 

52reshape = get_xp(np)(_aliases.reshape) 

53argsort = get_xp(np)(_aliases.argsort) 

54sort = get_xp(np)(_aliases.sort) 

55nonzero = get_xp(np)(_aliases.nonzero) 

56sum = get_xp(np)(_aliases.sum) 

57prod = get_xp(np)(_aliases.prod) 

58ceil = get_xp(np)(_aliases.ceil) 

59floor = get_xp(np)(_aliases.floor) 

60trunc = get_xp(np)(_aliases.trunc) 

61matmul = get_xp(np)(_aliases.matmul) 

62matrix_transpose = get_xp(np)(_aliases.matrix_transpose) 

63tensordot = get_xp(np)(_aliases.tensordot) 

64 

65# These functions are completely new here. If the library already has them 

66# (i.e., numpy 2.0), use the library version instead of our wrapper. 

67if hasattr(np, 'vecdot'): 

68 vecdot = np.vecdot 

69else: 

70 vecdot = get_xp(np)(_aliases.vecdot) 

71if hasattr(np, 'isdtype'): 

72 isdtype = np.isdtype 

73else: 

74 isdtype = get_xp(np)(_aliases.isdtype) 

75 

76__all__ = _aliases.__all__ + ['asarray', 'asarray_numpy', 'bool', 'acos', 

77 'acosh', 'asin', 'asinh', 'atan', 'atan2', 

78 'atanh', 'bitwise_left_shift', 'bitwise_invert', 

79 'bitwise_right_shift', 'concat', 'pow'] 

80 

81_all_ignore = ['np', 'get_xp']