histograms.pyi 1.3 KB

123456789101112131415161718192021222324252627
  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_i8: npt.NDArray[np.int64]
  6. AR_f8: npt.NDArray[np.float64]
  7. assert_type(np.histogram_bin_edges(AR_i8, bins="auto"), npt.NDArray[Any])
  8. assert_type(np.histogram_bin_edges(AR_i8, bins="rice", range=(0, 3)), npt.NDArray[Any])
  9. assert_type(np.histogram_bin_edges(AR_i8, bins="scott", weights=AR_f8), npt.NDArray[Any])
  10. assert_type(np.histogram(AR_i8, bins="auto"), tuple[npt.NDArray[Any], npt.NDArray[Any]])
  11. assert_type(np.histogram(AR_i8, bins="rice", range=(0, 3)), tuple[npt.NDArray[Any], npt.NDArray[Any]])
  12. assert_type(np.histogram(AR_i8, bins="scott", weights=AR_f8), tuple[npt.NDArray[Any], npt.NDArray[Any]])
  13. assert_type(np.histogram(AR_f8, bins=1, density=True), tuple[npt.NDArray[Any], npt.NDArray[Any]])
  14. assert_type(np.histogramdd(AR_i8, bins=[1]),
  15. tuple[npt.NDArray[Any], tuple[npt.NDArray[Any], ...]])
  16. assert_type(np.histogramdd(AR_i8, range=[(0, 3)]),
  17. tuple[npt.NDArray[Any], tuple[npt.NDArray[Any], ...]])
  18. assert_type(np.histogramdd(AR_i8, weights=AR_f8),
  19. tuple[npt.NDArray[Any], tuple[npt.NDArray[Any], ...]])
  20. assert_type(np.histogramdd(AR_f8, density=True),
  21. tuple[npt.NDArray[Any], tuple[npt.NDArray[Any], ...]])