emath.pyi 2.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556
  1. from typing import Any
  2. import numpy as np
  3. import numpy.typing as npt
  4. from typing_extensions import assert_type
  5. AR_f8: npt.NDArray[np.float64]
  6. AR_c16: npt.NDArray[np.complex128]
  7. f8: np.float64
  8. c16: np.complex128
  9. assert_type(np.emath.sqrt(f8), Any)
  10. assert_type(np.emath.sqrt(AR_f8), npt.NDArray[Any])
  11. assert_type(np.emath.sqrt(c16), np.complexfloating[Any, Any])
  12. assert_type(np.emath.sqrt(AR_c16), npt.NDArray[np.complexfloating[Any, Any]])
  13. assert_type(np.emath.log(f8), Any)
  14. assert_type(np.emath.log(AR_f8), npt.NDArray[Any])
  15. assert_type(np.emath.log(c16), np.complexfloating[Any, Any])
  16. assert_type(np.emath.log(AR_c16), npt.NDArray[np.complexfloating[Any, Any]])
  17. assert_type(np.emath.log10(f8), Any)
  18. assert_type(np.emath.log10(AR_f8), npt.NDArray[Any])
  19. assert_type(np.emath.log10(c16), np.complexfloating[Any, Any])
  20. assert_type(np.emath.log10(AR_c16), npt.NDArray[np.complexfloating[Any, Any]])
  21. assert_type(np.emath.log2(f8), Any)
  22. assert_type(np.emath.log2(AR_f8), npt.NDArray[Any])
  23. assert_type(np.emath.log2(c16), np.complexfloating[Any, Any])
  24. assert_type(np.emath.log2(AR_c16), npt.NDArray[np.complexfloating[Any, Any]])
  25. assert_type(np.emath.logn(f8, 2), Any)
  26. assert_type(np.emath.logn(AR_f8, 4), npt.NDArray[Any])
  27. assert_type(np.emath.logn(f8, 1j), np.complexfloating[Any, Any])
  28. assert_type(np.emath.logn(AR_c16, 1.5), npt.NDArray[np.complexfloating[Any, Any]])
  29. assert_type(np.emath.power(f8, 2), Any)
  30. assert_type(np.emath.power(AR_f8, 4), npt.NDArray[Any])
  31. assert_type(np.emath.power(f8, 2j), np.complexfloating[Any, Any])
  32. assert_type(np.emath.power(AR_c16, 1.5), npt.NDArray[np.complexfloating[Any, Any]])
  33. assert_type(np.emath.arccos(f8), Any)
  34. assert_type(np.emath.arccos(AR_f8), npt.NDArray[Any])
  35. assert_type(np.emath.arccos(c16), np.complexfloating[Any, Any])
  36. assert_type(np.emath.arccos(AR_c16), npt.NDArray[np.complexfloating[Any, Any]])
  37. assert_type(np.emath.arcsin(f8), Any)
  38. assert_type(np.emath.arcsin(AR_f8), npt.NDArray[Any])
  39. assert_type(np.emath.arcsin(c16), np.complexfloating[Any, Any])
  40. assert_type(np.emath.arcsin(AR_c16), npt.NDArray[np.complexfloating[Any, Any]])
  41. assert_type(np.emath.arctanh(f8), Any)
  42. assert_type(np.emath.arctanh(AR_f8), npt.NDArray[Any])
  43. assert_type(np.emath.arctanh(c16), np.complexfloating[Any, Any])
  44. assert_type(np.emath.arctanh(AR_c16), npt.NDArray[np.complexfloating[Any, Any]])