mod.py 8.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149
  1. import numpy as np
  2. f8 = np.float64()
  3. i8 = np.int64()
  4. u8 = np.uint64()
  5. f4 = np.float32()
  6. i4 = np.int32()
  7. u4 = np.uint32()
  8. td = np.timedelta64(0, "D")
  9. b_ = np.bool_()
  10. b = bool()
  11. f = float()
  12. i = int()
  13. AR = np.array([1], dtype=np.bool_)
  14. AR.setflags(write=False)
  15. AR2 = np.array([1], dtype=np.timedelta64)
  16. AR2.setflags(write=False)
  17. # Time structures
  18. reveal_type(td % td) # E: numpy.timedelta64
  19. reveal_type(AR2 % td) # E: Any
  20. reveal_type(td % AR2) # E: Any
  21. reveal_type(divmod(td, td)) # E: Tuple[numpy.signedinteger[numpy.typing._64Bit], numpy.timedelta64]
  22. reveal_type(divmod(AR2, td)) # E: Tuple[Any, Any]
  23. reveal_type(divmod(td, AR2)) # E: Tuple[Any, Any]
  24. # Bool
  25. reveal_type(b_ % b) # E: numpy.signedinteger[numpy.typing._8Bit]
  26. reveal_type(b_ % i) # E: numpy.signedinteger[Any]
  27. reveal_type(b_ % f) # E: numpy.floating[numpy.typing._64Bit]
  28. reveal_type(b_ % b_) # E: numpy.signedinteger[numpy.typing._8Bit]
  29. reveal_type(b_ % i8) # E: numpy.signedinteger[numpy.typing._64Bit]
  30. reveal_type(b_ % u8) # E: numpy.unsignedinteger[numpy.typing._64Bit]
  31. reveal_type(b_ % f8) # E: numpy.floating[numpy.typing._64Bit]
  32. reveal_type(b_ % AR) # E: Any
  33. reveal_type(divmod(b_, b)) # E: Tuple[numpy.signedinteger[numpy.typing._8Bit], numpy.signedinteger[numpy.typing._8Bit]]
  34. reveal_type(divmod(b_, i)) # E: Tuple[numpy.signedinteger[Any], numpy.signedinteger[Any]]
  35. reveal_type(divmod(b_, f)) # E: Tuple[numpy.floating[numpy.typing._64Bit], numpy.floating[numpy.typing._64Bit]]
  36. reveal_type(divmod(b_, b_)) # E: Tuple[numpy.signedinteger[numpy.typing._8Bit], numpy.signedinteger[numpy.typing._8Bit]]
  37. reveal_type(divmod(b_, i8)) # E: Tuple[numpy.signedinteger[numpy.typing._64Bit], numpy.signedinteger[numpy.typing._64Bit]]
  38. reveal_type(divmod(b_, u8)) # E: Tuple[numpy.unsignedinteger[numpy.typing._64Bit], numpy.unsignedinteger[numpy.typing._64Bit]]
  39. reveal_type(divmod(b_, f8)) # E: Tuple[numpy.floating[numpy.typing._64Bit], numpy.floating[numpy.typing._64Bit]]
  40. reveal_type(divmod(b_, AR)) # E: Tuple[Any, Any]
  41. reveal_type(b % b_) # E: numpy.signedinteger[numpy.typing._8Bit]
  42. reveal_type(i % b_) # E: numpy.signedinteger[Any]
  43. reveal_type(f % b_) # E: numpy.floating[numpy.typing._64Bit]
  44. reveal_type(b_ % b_) # E: numpy.signedinteger[numpy.typing._8Bit]
  45. reveal_type(i8 % b_) # E: numpy.signedinteger[numpy.typing._64Bit]
  46. reveal_type(u8 % b_) # E: numpy.unsignedinteger[numpy.typing._64Bit]
  47. reveal_type(f8 % b_) # E: numpy.floating[numpy.typing._64Bit]
  48. reveal_type(AR % b_) # E: Any
  49. reveal_type(divmod(b, b_)) # E: Tuple[numpy.signedinteger[numpy.typing._8Bit], numpy.signedinteger[numpy.typing._8Bit]]
  50. reveal_type(divmod(i, b_)) # E: Tuple[numpy.signedinteger[Any], numpy.signedinteger[Any]]
  51. reveal_type(divmod(f, b_)) # E: Tuple[numpy.floating[numpy.typing._64Bit], numpy.floating[numpy.typing._64Bit]]
  52. reveal_type(divmod(b_, b_)) # E: Tuple[numpy.signedinteger[numpy.typing._8Bit], numpy.signedinteger[numpy.typing._8Bit]]
  53. reveal_type(divmod(i8, b_)) # E: Tuple[numpy.signedinteger[numpy.typing._64Bit], numpy.signedinteger[numpy.typing._64Bit]]
  54. reveal_type(divmod(u8, b_)) # E: Tuple[numpy.unsignedinteger[numpy.typing._64Bit], numpy.unsignedinteger[numpy.typing._64Bit]]
  55. reveal_type(divmod(f8, b_)) # E: Tuple[numpy.floating[numpy.typing._64Bit], numpy.floating[numpy.typing._64Bit]]
  56. reveal_type(divmod(AR, b_)) # E: Tuple[Any, Any]
  57. # int
  58. reveal_type(i8 % b) # E: numpy.signedinteger[numpy.typing._64Bit]
  59. reveal_type(i8 % i) # E: numpy.signedinteger[Any]
  60. reveal_type(i8 % f) # E: numpy.floating[numpy.typing._64Bit]
  61. reveal_type(i8 % i8) # E: numpy.signedinteger[numpy.typing._64Bit]
  62. reveal_type(i8 % f8) # E: numpy.floating[numpy.typing._64Bit]
  63. reveal_type(i4 % i8) # E: numpy.signedinteger[numpy.typing._64Bit]
  64. reveal_type(i4 % f8) # E: numpy.floating[numpy.typing._64Bit]
  65. reveal_type(i4 % i4) # E: numpy.signedinteger[numpy.typing._32Bit]
  66. reveal_type(i4 % f4) # E: numpy.floating[numpy.typing._32Bit]
  67. reveal_type(i8 % AR) # E: Any
  68. reveal_type(divmod(i8, b)) # E: Tuple[numpy.signedinteger[numpy.typing._64Bit], numpy.signedinteger[numpy.typing._64Bit]]
  69. reveal_type(divmod(i8, i)) # E: Tuple[numpy.signedinteger[Any], numpy.signedinteger[Any]]
  70. reveal_type(divmod(i8, f)) # E: Tuple[numpy.floating[numpy.typing._64Bit], numpy.floating[numpy.typing._64Bit]]
  71. reveal_type(divmod(i8, i8)) # E: Tuple[numpy.signedinteger[numpy.typing._64Bit], numpy.signedinteger[numpy.typing._64Bit]]
  72. reveal_type(divmod(i8, f8)) # E: Tuple[numpy.floating[numpy.typing._64Bit], numpy.floating[numpy.typing._64Bit]]
  73. reveal_type(divmod(i8, i4)) # E: Tuple[numpy.signedinteger[numpy.typing._64Bit], numpy.signedinteger[numpy.typing._64Bit]]
  74. reveal_type(divmod(i8, f4)) # E: Tuple[numpy.floating[numpy.typing._64Bit], numpy.floating[numpy.typing._64Bit]]
  75. reveal_type(divmod(i4, i4)) # E: Tuple[numpy.signedinteger[numpy.typing._32Bit], numpy.signedinteger[numpy.typing._32Bit]]
  76. reveal_type(divmod(i4, f4)) # E: Tuple[numpy.floating[numpy.typing._32Bit], numpy.floating[numpy.typing._32Bit]]
  77. reveal_type(divmod(i8, AR)) # E: Tuple[Any, Any]
  78. reveal_type(b % i8) # E: numpy.signedinteger[numpy.typing._64Bit]
  79. reveal_type(i % i8) # E: numpy.signedinteger[Any]
  80. reveal_type(f % i8) # E: numpy.floating[numpy.typing._64Bit]
  81. reveal_type(i8 % i8) # E: numpy.signedinteger[numpy.typing._64Bit]
  82. reveal_type(f8 % i8) # E: numpy.floating[numpy.typing._64Bit]
  83. reveal_type(i8 % i4) # E: numpy.signedinteger[numpy.typing._64Bit]
  84. reveal_type(f8 % i4) # E: numpy.floating[numpy.typing._64Bit]
  85. reveal_type(i4 % i4) # E: numpy.signedinteger[numpy.typing._32Bit]
  86. reveal_type(f4 % i4) # E: numpy.floating[numpy.typing._32Bit]
  87. reveal_type(AR % i8) # E: Any
  88. reveal_type(divmod(b, i8)) # E: Tuple[numpy.signedinteger[numpy.typing._64Bit], numpy.signedinteger[numpy.typing._64Bit]]
  89. reveal_type(divmod(i, i8)) # E: Tuple[numpy.signedinteger[Any], numpy.signedinteger[Any]]
  90. reveal_type(divmod(f, i8)) # E: Tuple[numpy.floating[numpy.typing._64Bit], numpy.floating[numpy.typing._64Bit]]
  91. reveal_type(divmod(i8, i8)) # E: Tuple[numpy.signedinteger[numpy.typing._64Bit], numpy.signedinteger[numpy.typing._64Bit]]
  92. reveal_type(divmod(f8, i8)) # E: Tuple[numpy.floating[numpy.typing._64Bit], numpy.floating[numpy.typing._64Bit]]
  93. reveal_type(divmod(i4, i8)) # E: Tuple[numpy.signedinteger[numpy.typing._64Bit], numpy.signedinteger[numpy.typing._64Bit]]
  94. reveal_type(divmod(f4, i8)) # E: Tuple[numpy.floating[numpy.typing._64Bit], numpy.floating[numpy.typing._64Bit]]
  95. reveal_type(divmod(i4, i4)) # E: Tuple[numpy.signedinteger[numpy.typing._32Bit], numpy.signedinteger[numpy.typing._32Bit]]
  96. reveal_type(divmod(f4, i4)) # E: Tuple[numpy.floating[numpy.typing._32Bit], numpy.floating[numpy.typing._32Bit]]
  97. reveal_type(divmod(AR, i8)) # E: Tuple[Any, Any]
  98. # float
  99. reveal_type(f8 % b) # E: numpy.floating[numpy.typing._64Bit]
  100. reveal_type(f8 % i) # E: numpy.floating[Any]
  101. reveal_type(f8 % f) # E: numpy.floating[numpy.typing._64Bit]
  102. reveal_type(i8 % f4) # E: numpy.floating[numpy.typing._64Bit]
  103. reveal_type(f4 % f4) # E: numpy.floating[numpy.typing._32Bit]
  104. reveal_type(f8 % AR) # E: Any
  105. reveal_type(divmod(f8, b)) # E: Tuple[numpy.floating[numpy.typing._64Bit], numpy.floating[numpy.typing._64Bit]]
  106. reveal_type(divmod(f8, i)) # E: Tuple[numpy.floating[Any], numpy.floating[Any]]
  107. reveal_type(divmod(f8, f)) # E: Tuple[numpy.floating[numpy.typing._64Bit], numpy.floating[numpy.typing._64Bit]]
  108. reveal_type(divmod(f8, f8)) # E: Tuple[numpy.floating[numpy.typing._64Bit], numpy.floating[numpy.typing._64Bit]]
  109. reveal_type(divmod(f8, f4)) # E: Tuple[numpy.floating[numpy.typing._64Bit], numpy.floating[numpy.typing._64Bit]]
  110. reveal_type(divmod(f4, f4)) # E: Tuple[numpy.floating[numpy.typing._32Bit], numpy.floating[numpy.typing._32Bit]]
  111. reveal_type(divmod(f8, AR)) # E: Tuple[Any, Any]
  112. reveal_type(b % f8) # E: numpy.floating[numpy.typing._64Bit]
  113. reveal_type(i % f8) # E: numpy.floating[Any]
  114. reveal_type(f % f8) # E: numpy.floating[numpy.typing._64Bit]
  115. reveal_type(f8 % f8) # E: numpy.floating[numpy.typing._64Bit]
  116. reveal_type(f8 % f8) # E: numpy.floating[numpy.typing._64Bit]
  117. reveal_type(f4 % f4) # E: numpy.floating[numpy.typing._32Bit]
  118. reveal_type(AR % f8) # E: Any
  119. reveal_type(divmod(b, f8)) # E: Tuple[numpy.floating[numpy.typing._64Bit], numpy.floating[numpy.typing._64Bit]]
  120. reveal_type(divmod(i, f8)) # E: Tuple[numpy.floating[Any], numpy.floating[Any]]
  121. reveal_type(divmod(f, f8)) # E: Tuple[numpy.floating[numpy.typing._64Bit], numpy.floating[numpy.typing._64Bit]]
  122. reveal_type(divmod(f8, f8)) # E: Tuple[numpy.floating[numpy.typing._64Bit], numpy.floating[numpy.typing._64Bit]]
  123. reveal_type(divmod(f4, f8)) # E: Tuple[numpy.floating[numpy.typing._64Bit], numpy.floating[numpy.typing._64Bit]]
  124. reveal_type(divmod(f4, f4)) # E: Tuple[numpy.floating[numpy.typing._32Bit], numpy.floating[numpy.typing._32Bit]]
  125. reveal_type(divmod(AR, f8)) # E: Tuple[Any, Any]