math.lua 8.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288
  1. print("testing numbers and math lib")
  2. -- basic float notation
  3. assert(0e12 == 0 and .0 == 0 and 0. == 0 and .2e2 == 20 and 2.E-1 == 0.2)
  4. do
  5. local a, b, c = "2", " 3e0 ", " 10 "
  6. assert(a + b == 5 and -b == -3 and b + "2" == 5 and "10" - c == 0)
  7. assert(type(a) == 'string' and type(b) == 'string' and type(c) == 'string')
  8. assert(a == "2" and b == " 3e0 " and c == " 10 " and -c == -" 10 ")
  9. assert(c % a == 0 and a ^ b == 08)
  10. a = 0
  11. assert(a == -a and 0 == -0)
  12. end
  13. do
  14. local x = -1
  15. local mz = 0 / x -- minus zero
  16. t = { [0] = 10, 20, 30, 40, 50 }
  17. assert(t[mz] == t[0] and t[-0] == t[0])
  18. end
  19. do
  20. local a, b = math.modf(3.5)
  21. assert(a == 3 and b == 0.5)
  22. assert(math.huge > 10e30)
  23. assert(-math.huge < -10e30)
  24. end
  25. function f(...)
  26. if select('#', ...) == 1 then
  27. return (...)
  28. else
  29. return "***"
  30. end
  31. end
  32. -- testing numeric strings
  33. assert("2" + 1 == 3)
  34. assert("2 " + 1 == 3)
  35. assert(" -2 " + 1 == -1)
  36. assert(" -0xa " + 1 == -9)
  37. -- testing 'tonumber'
  38. assert(tonumber {} == nil)
  39. assert(tonumber '+0.01' == 1 / 100 and tonumber '+.01' == 0.01 and
  40. tonumber '.01' == 0.01 and tonumber '-1.' == -1 and
  41. tonumber '+1.' == 1)
  42. assert(tonumber '+ 0.01' == nil and tonumber '+.e1' == nil and
  43. tonumber '1e' == nil and tonumber '1.0e+' == nil and
  44. tonumber '.' == nil)
  45. assert(tonumber('-012') == -010 - 2)
  46. assert(tonumber('-1.2e2') == - - -120)
  47. assert(tonumber("0xffffffffffff") == 2 ^ (4 * 12) - 1)
  48. -- assert(tonumber("0x" .. string.rep("f", 150)) == 2 ^ (4 * 150) - 1)
  49. assert(tonumber('0x3.' .. string.rep('0', 100)) == 3)
  50. assert(tonumber('0x0.' .. string.rep('0', 150) .. "1") == 2 ^ (-4 * 151))
  51. -- testing 'tonumber' with base
  52. assert(tonumber(' 001010 ', 2) == 10)
  53. assert(tonumber(' 001010 ', 10) == 001010)
  54. assert(tonumber(' -1010 ', 2) == -10)
  55. assert(tonumber('10', 36) == 36)
  56. assert(tonumber(' -10 ', 36) == -36)
  57. assert(tonumber(' +1Z ', 36) == 36 + 35)
  58. assert(tonumber(' -1z ', 36) == -36 + -35)
  59. assert(tonumber('-fFfa', 16) == -(10 + (16 * (15 + (16 * (15 + (16 * 15)))))))
  60. assert(tonumber(string.rep('1', 42), 2) + 1 == 2 ^ 42)
  61. assert(tonumber(string.rep('1', 34), 2) + 1 == 2 ^ 34)
  62. assert(tonumber('ffffFFFF', 16) + 1 == 2 ^ 32)
  63. assert(tonumber('0ffffFFFF', 16) + 1 == 2 ^ 32)
  64. -- assert(tonumber('-0ffffffFFFF', 16) - 1 == -2 ^ 40)
  65. for i = 2, 36 do
  66. assert(tonumber('\t10000000000\t', i) == i ^ 10)
  67. end
  68. -- testing 'tonumber' fo invalid formats
  69. assert(f(tonumber('fFfa', 15)) == nil)
  70. assert(f(tonumber('099', 8)) == nil)
  71. -- assert(f(tonumber('1\0', 2)) == nil)
  72. assert(f(tonumber('', 8)) == nil)
  73. assert(f(tonumber(' ', 9)) == nil)
  74. assert(f(tonumber(' ', 9)) == nil)
  75. assert(f(tonumber('0xf', 10)) == nil)
  76. assert(f(tonumber('inf')) == nil)
  77. assert(f(tonumber(' INF ')) == nil)
  78. assert(f(tonumber('Nan')) == nil)
  79. assert(f(tonumber('nan')) == nil)
  80. assert(f(tonumber(' ')) == nil)
  81. assert(f(tonumber('')) == nil)
  82. assert(f(tonumber('1 a')) == nil)
  83. -- assert(f(tonumber('1\0')) == nil)
  84. -- assert(f(tonumber('1 \0')) == nil)
  85. -- assert(f(tonumber('1\0 ')) == nil)
  86. assert(f(tonumber('e1')) == nil)
  87. assert(f(tonumber('e 1')) == nil)
  88. assert(f(tonumber(' 3.4.5 ')) == nil)
  89. -- testing 'tonumber' for invalid hexadecimal formats
  90. assert(tonumber('0x') == nil)
  91. assert(tonumber('x') == nil)
  92. assert(tonumber('x3') == nil)
  93. assert(tonumber('00x2') == nil)
  94. assert(tonumber('0x 2') == nil)
  95. assert(tonumber('0 x2') == nil)
  96. assert(tonumber('23x') == nil)
  97. assert(tonumber('- 0xaa') == nil)
  98. -- testing hexadecimal numerals
  99. assert(0x10 == 16 and 0xfff == 2 ^ 12 - 1 and 0XFB == 251)
  100. assert(0x0p12 == 0 and 0x.0p-3 == 0)
  101. assert(0xFFFFFFFF == 2 ^ 32 - 1)
  102. assert(tonumber('+0x2') == 2)
  103. assert(tonumber('-0xaA') == -170)
  104. -- assert(tonumber('-0xffFFFfff') == -2 ^ 32 + 1)
  105. -- possible confusion with decimal exponent
  106. assert(0E+1 == 0 and 0xE + 1 == 15 and 0xe - 1 == 13)
  107. -- floating hexas
  108. assert(tonumber(' 0x2.5 ') == 0x25 / 16)
  109. assert(tonumber(' -0x2.5 ') == -0x25 / 16)
  110. assert(tonumber(' +0x0.51p+8 ') == 0x51)
  111. -- assert(tonumber('0x0.51p') == nil)
  112. -- assert(tonumber('0x5p+-2') == nil)
  113. assert(0x.FfffFFFF == 1 - '0x.00000001')
  114. assert('0xA.a' + 0 == 10 + 10 / 16)
  115. assert(0xa.aP4 == 0XAA)
  116. print(0x4P-2)
  117. assert(0x4P-2 == 1)
  118. assert(0x1.1 == '0x1.' + '+0x.1')
  119. assert(1.1 == 1. + .1)
  120. assert(100.0 == 1E2 and .01 == 1e-2)
  121. assert(1111111111111111 - 1111111111111110 == 1000.00e-03)
  122. -- 1234567890123456
  123. assert(1.1 == '1.' + '.1')
  124. assert('1111111111111111' - '1111111111111110' == tonumber " +0.001e+3 \n\t")
  125. function eq(a, b, limit)
  126. if not limit then limit = 10E-10 end
  127. return math.abs(a - b) <= limit
  128. end
  129. assert(0.1e-30 > 0.9E-31 and 0.9E30 < 0.1e31)
  130. assert(0.123456 > 0.123455)
  131. assert(tonumber('+1.23E18') == 1.23 * 10 ^ 18)
  132. -- testing order operators
  133. assert(not (1 < 1) and (1 < 2) and not (2 < 1))
  134. assert(not ('a' < 'a') and ('a' < 'b') and not ('b' < 'a'))
  135. assert((1 <= 1) and (1 <= 2) and not (2 <= 1))
  136. assert(('a' <= 'a') and ('a' <= 'b') and not ('b' <= 'a'))
  137. print(not (1 > 1), not (1 > 2), 2 > 1)
  138. assert(not (1 > 1) and not (1 > 2) and (2 > 1))
  139. assert(not ('a' > 'a') and not ('a' > 'b') and ('b' > 'a'))
  140. assert((1 >= 1) and not (1 >= 2) and (2 >= 1))
  141. assert(('a' >= 'a') and not ('a' >= 'b') and ('b' >= 'a'))
  142. -- testing mod operator
  143. assert(-4 % 3 == 2)
  144. assert(4 % -3 == -2)
  145. assert(math.pi - math.pi % 1 == 3)
  146. assert(math.pi - math.pi % 0.001 == 3.141)
  147. local function testbit(a, n)
  148. return a / 2 ^ n % 2 >= 1
  149. end
  150. assert(eq(math.sin(-9.8) ^ 2 + math.cos(-9.8) ^ 2, 1))
  151. assert(eq(math.tan(math.pi / 4), 1))
  152. assert(eq(math.sin(math.pi / 2), 1) and eq(math.cos(math.pi / 2), 0))
  153. assert(eq(math.atan(1), math.pi / 4) and eq(math.acos(0), math.pi / 2) and
  154. eq(math.asin(1), math.pi / 2))
  155. assert(eq(math.deg(math.pi / 2), 90) and eq(math.rad(90), math.pi / 2))
  156. assert(math.abs(-10) == 10)
  157. assert(eq(math.atan2(1, 0), math.pi / 2))
  158. assert(math.ceil(4.5) == 5.0)
  159. assert(math.floor(4.5) == 4.0)
  160. assert(math.fmod(10, 3) == 1)
  161. assert(eq(math.sqrt(10) ^ 2, 10))
  162. assert(eq(math.log(2, 10), math.log(2) / math.log(10)))
  163. assert(eq(math.log(2, 2), 1))
  164. assert(eq(math.log(9, 3), 2))
  165. assert(eq(math.exp(0), 1))
  166. assert(eq(math.sin(10), math.sin(10 % (2 * math.pi))))
  167. local v, e = math.frexp(math.pi)
  168. assert(eq(math.ldexp(v, e), math.pi))
  169. assert(eq(math.tanh(3.5), math.sinh(3.5) / math.cosh(3.5)))
  170. assert(tonumber(' 1.3e-2 ') == 1.3e-2)
  171. assert(tonumber(' -1.00000000000001 ') == -1.00000000000001)
  172. -- testing constant limits
  173. -- 2^23 = 8388608
  174. assert(8388609 + -8388609 == 0)
  175. assert(8388608 + -8388608 == 0)
  176. assert(8388607 + -8388607 == 0)
  177. -- testing implicit convertions
  178. local a, b = '10', '20'
  179. assert(a * b == 200 and a + b == 30 and a - b == -10 and a / b == 0.5 and -b == -20)
  180. assert(a == '10' and b == '20')
  181. if not _port then
  182. print("testing -0 and NaN")
  183. local mz, z = -0, 0
  184. assert(mz == z)
  185. -- assert(1 / mz < 0 and 0 < 1 / z)
  186. local a = { [mz] = 1 }
  187. assert(a[z] == 1 and a[mz] == 1)
  188. local inf = math.huge * 2 + 1
  189. mz, z = -1 / inf, 1 / inf
  190. assert(mz == z)
  191. assert(1 / mz < 0 and 0 < 1 / z)
  192. local NaN = inf - inf
  193. assert(NaN ~= NaN)
  194. assert(not (NaN < NaN))
  195. assert(not (NaN <= NaN))
  196. assert(not (NaN > NaN))
  197. assert(not (NaN >= NaN))
  198. assert(not (0 < NaN) and not (NaN < 0))
  199. local NaN1 = 0 / 0
  200. assert(NaN ~= NaN1 and not (NaN <= NaN1) and not (NaN1 <= NaN))
  201. local a = {}
  202. assert(not pcall(function() a[NaN] = 1 end))
  203. assert(a[NaN] == nil)
  204. a[1] = 1
  205. assert(not pcall(function() a[NaN] = 1 end))
  206. assert(a[NaN] == nil)
  207. -- string with same binary representation as 0.0 (may create problems
  208. -- for constant manipulation in the pre-compiler)
  209. local a1, a2, a3, a4, a5 = 0, 0, "\0\0\0\0\0\0\0\0", 0, "\0\0\0\0\0\0\0\0"
  210. assert(a1 == a2 and a2 == a4 and a1 ~= a3)
  211. assert(a3 == a5)
  212. end
  213. -- if not _port then
  214. -- print("testing 'math.random'")
  215. -- math.randomseed(0)
  216. -- local function aux(x1, x2, p)
  217. -- local Max = -math.huge
  218. -- local Min = math.huge
  219. -- for i = 0, 20000 do
  220. -- local t = math.random(table.unpack(p))
  221. -- Max = math.max(Max, t)
  222. -- Min = math.min(Min, t)
  223. -- if eq(Max, x2, 0.001) and eq(Min, x1, 0.001) then
  224. -- goto ok
  225. -- end
  226. -- end
  227. -- -- loop ended without satisfing condition
  228. -- assert(false)
  229. -- ::ok::
  230. -- assert(x1 <= Min and Max <= x2)
  231. -- end
  232. -- aux(0, 1, {})
  233. -- aux(-10, 0, { -10, 0 })
  234. -- end
  235. for i = 1, 10 do
  236. local t = math.random(5)
  237. assert(1 <= t and t <= 5)
  238. end
  239. print('OK')