NTest_ws2812.lua 8.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262
  1. local N = ...
  2. N = (N or require "NTest")("ws2812 buffers")
  3. local buffer, buffer1, buffer2
  4. local function initBuffer(buf, ...)
  5. for i,v in ipairs({...}) do
  6. buf:set(i, v, v*2, v*3, v*4)
  7. end
  8. return buf
  9. end
  10. local function equalsBuffer(buf1, buf2)
  11. return eq(buf1:dump(), buf2:dump())
  12. end
  13. N.test('initialize a buffer', function()
  14. buffer = ws2812.newBuffer(9, 3)
  15. nok(buffer == nil)
  16. ok(eq(buffer:size(), 9), "check size")
  17. ok(eq(buffer:dump(), string.char(0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0)), "initialize with 0")
  18. fail(function() ws2812.newBuffer(9, 0) end, "should be a positive integer")
  19. fail(function() ws2812.newBuffer(9, -1) end, "should be a positive integer")
  20. fail(function() ws2812.newBuffer(0, 3) end, "should be a positive integer")
  21. fail(function() ws2812.newBuffer(-1, 3) end, "should be a positive integer")
  22. end)
  23. N.test('have correct size', function()
  24. buffer = ws2812.newBuffer(9, 3)
  25. ok(eq(buffer:size(), 9), "check size")
  26. buffer = ws2812.newBuffer(9, 22)
  27. ok(eq(buffer:size(), 9), "check size")
  28. buffer = ws2812.newBuffer(13, 1)
  29. ok(eq(buffer:size(), 13), "check size")
  30. end)
  31. N.test('fill a buffer with one color', function()
  32. buffer = ws2812.newBuffer(3, 3)
  33. buffer:fill(1,222,55)
  34. ok(eq(buffer:dump(), string.char(1,222,55,1,222,55,1,222,55)), "RGB")
  35. buffer = ws2812.newBuffer(3, 4)
  36. buffer:fill(1,222,55, 77)
  37. ok(eq(buffer:dump(), string.char(1,222,55,77,1,222,55,77,1,222,55,77)), "RGBW")
  38. end)
  39. N.test('replace correctly', function()
  40. buffer = ws2812.newBuffer(5, 3)
  41. buffer:replace(string.char(3,255,165,33,0,244,12,87,255))
  42. ok(eq(buffer:dump(), string.char(3,255,165,33,0,244,12,87,255,0,0,0,0,0,0)), "RGBW")
  43. buffer = ws2812.newBuffer(5, 3)
  44. buffer:replace(string.char(3,255,165,33,0,244,12,87,255), 2)
  45. ok(eq(buffer:dump(), string.char(0,0,0,3,255,165,33,0,244,12,87,255,0,0,0)), "RGBW")
  46. buffer = ws2812.newBuffer(5, 3)
  47. buffer:replace(string.char(3,255,165,33,0,244,12,87,255), -5)
  48. ok(eq(buffer:dump(), string.char(3,255,165,33,0,244,12,87,255,0,0,0,0,0,0)), "RGBW")
  49. fail(function() buffer:replace(string.char(3,255,165,33,0,244,12,87,255), 4) end, "does not fit into destination")
  50. end)
  51. N.test('replace correctly issue #2921', function()
  52. buffer = ws2812.newBuffer(5, 3)
  53. buffer:replace(string.char(3,255,165,33,0,244,12,87,255), -7)
  54. ok(eq(buffer:dump(), string.char(3,255,165,33,0,244,12,87,255,0,0,0,0,0,0)), "RGBW")
  55. end)
  56. N.test('get/set correctly', function()
  57. buffer = ws2812.newBuffer(3, 4)
  58. buffer:fill(1,222,55,13)
  59. ok(eq({buffer:get(2)},{1,222,55,13}))
  60. buffer:set(2, 4,53,99,0)
  61. ok(eq({buffer:get(1)},{1,222,55,13}))
  62. ok(eq({buffer:get(2)},{4,53,99,0}))
  63. ok(eq(buffer:dump(), string.char(1,222,55,13,4,53,99,0,1,222,55,13)), "RGBW")
  64. fail(function() buffer:get(0) end, "index out of range")
  65. fail(function() buffer:get(4) end, "index out of range")
  66. fail(function() buffer:set(0,1,2,3,4) end, "index out of range")
  67. fail(function() buffer:set(4,1,2,3,4) end, "index out of range")
  68. fail(function() buffer:set(2,1,2,3) end, "number expected, got no value")
  69. -- fail(function() buffer:set(2,1,2,3,4,5) end, "extra values given")
  70. end)
  71. N.test('fade correctly', function()
  72. buffer = ws2812.newBuffer(1, 3)
  73. buffer:fill(1,222,55)
  74. buffer:fade(2)
  75. ok(buffer:dump() == string.char(0,111,27), "RGB")
  76. buffer:fill(1,222,55)
  77. buffer:fade(3, ws2812.FADE_OUT)
  78. ok(buffer:dump() == string.char(0,math.floor(222/3),math.floor(55/3)), "RGB")
  79. buffer:fill(1,222,55)
  80. buffer:fade(3, ws2812.FADE_IN)
  81. ok(buffer:dump() == string.char(3,255,165), "RGB")
  82. buffer = ws2812.newBuffer(1, 4)
  83. buffer:fill(1,222,55, 77)
  84. buffer:fade(2, ws2812.FADE_OUT)
  85. ok(eq(buffer:dump(), string.char(0,111,27,38)), "RGBW")
  86. end)
  87. N.test('mix correctly issue #1736', function()
  88. buffer1 = ws2812.newBuffer(1, 3)
  89. buffer2 = ws2812.newBuffer(1, 3)
  90. buffer1:fill(10,22,54)
  91. buffer2:fill(10,27,55)
  92. buffer1:mix(256/8*7,buffer1,256/8,buffer2)
  93. ok(eq({buffer1:get(1)}, {10,23,54}))
  94. end)
  95. N.test('mix saturation correctly ', function()
  96. buffer1 = ws2812.newBuffer(1, 3)
  97. buffer2 = ws2812.newBuffer(1, 3)
  98. buffer1:fill(10,22,54)
  99. buffer2:fill(10,27,55)
  100. buffer1:mix(256/2,buffer1,-256,buffer2)
  101. ok(eq({buffer1:get(1)}, {0,0,0}))
  102. buffer1:fill(10,22,54)
  103. buffer2:fill(10,27,55)
  104. buffer1:mix(25600,buffer1,256/8,buffer2)
  105. ok(eq({buffer1:get(1)}, {255,255,255}))
  106. buffer1:fill(10,22,54)
  107. buffer2:fill(10,27,55)
  108. buffer1:mix(-257,buffer1,255,buffer2)
  109. ok(eq({buffer1:get(1)}, {0,5,1}))
  110. end)
  111. N.test('power', function()
  112. buffer = ws2812.newBuffer(2, 4)
  113. buffer:fill(10,22,54,234)
  114. ok(eq(buffer:power(), 2*(10+22+54+234)))
  115. end)
  116. N.test('shift LOGICAL', function()
  117. buffer1 = ws2812.newBuffer(4, 4)
  118. buffer2 = ws2812.newBuffer(4, 4)
  119. initBuffer(buffer1,7,8,9,12)
  120. initBuffer(buffer2,0,0,7,8)
  121. buffer1:shift(2)
  122. ok(equalsBuffer(buffer1, buffer2), "shift right")
  123. initBuffer(buffer1,7,8,9,12)
  124. initBuffer(buffer2,9,12,0,0)
  125. buffer1:shift(-2)
  126. ok(equalsBuffer(buffer1, buffer2), "shift left")
  127. initBuffer(buffer1,7,8,9,12)
  128. initBuffer(buffer2,7,0,8,12)
  129. buffer1:shift(1, nil, 2,3)
  130. ok(equalsBuffer(buffer1, buffer2), "shift middle right")
  131. initBuffer(buffer1,7,8,9,12)
  132. initBuffer(buffer2,7,9,0,12)
  133. buffer1:shift(-1, nil, 2,3)
  134. ok(equalsBuffer(buffer1, buffer2), "shift middle left")
  135. -- bounds checks, handle gracefully as string:sub does
  136. initBuffer(buffer1,7,8,9,12)
  137. initBuffer(buffer2,8,9,12,0)
  138. buffer1:shift(-1, ws2812.SHIFT_LOGICAL, 0,5)
  139. ok(equalsBuffer(buffer1, buffer2), "shift left out of bound")
  140. initBuffer(buffer1,7,8,9,12)
  141. initBuffer(buffer2,0,7,8,9)
  142. buffer1:shift(1, ws2812.SHIFT_LOGICAL, 0,5)
  143. ok(equalsBuffer(buffer1, buffer2), "shift right out of bound")
  144. end)
  145. N.test('shift LOGICAL issue #2946', function()
  146. buffer1 = ws2812.newBuffer(4, 4)
  147. buffer2 = ws2812.newBuffer(4, 4)
  148. initBuffer(buffer1,7,8,9,12)
  149. initBuffer(buffer2,0,0,0,0)
  150. buffer1:shift(4)
  151. ok(equalsBuffer(buffer1, buffer2), "shift all right")
  152. initBuffer(buffer1,7,8,9,12)
  153. initBuffer(buffer2,0,0,0,0)
  154. buffer1:shift(-4)
  155. ok(equalsBuffer(buffer1, buffer2), "shift all left")
  156. fail(function() buffer1:shift(10) end, "shifting more elements than buffer size")
  157. fail(function() buffer1:shift(-6) end, "shifting more elements than buffer size")
  158. end)
  159. N.test('shift CIRCULAR', function()
  160. buffer1 = ws2812.newBuffer(4, 4)
  161. buffer2 = ws2812.newBuffer(4, 4)
  162. initBuffer(buffer1,7,8,9,12)
  163. initBuffer(buffer2,9,12,7,8)
  164. buffer1:shift(2, ws2812.SHIFT_CIRCULAR)
  165. ok(equalsBuffer(buffer1, buffer2), "shift right")
  166. initBuffer(buffer1,7,8,9,12)
  167. initBuffer(buffer2,9,12,7,8)
  168. buffer1:shift(-2, ws2812.SHIFT_CIRCULAR)
  169. ok(equalsBuffer(buffer1, buffer2), "shift left")
  170. initBuffer(buffer1,7,8,9,12)
  171. initBuffer(buffer2,7,9,8,12)
  172. buffer1:shift(1, ws2812.SHIFT_CIRCULAR, 2,3)
  173. ok(equalsBuffer(buffer1, buffer2), "shift middle right")
  174. initBuffer(buffer1,7,8,9,12)
  175. initBuffer(buffer2,7,9,8,12)
  176. buffer1:shift(-1, ws2812.SHIFT_CIRCULAR, 2,3)
  177. ok(equalsBuffer(buffer1, buffer2), "shift middle left")
  178. -- bounds checks, handle gracefully as string:sub does
  179. initBuffer(buffer1,7,8,9,12)
  180. initBuffer(buffer2,8,9,12,7)
  181. buffer1:shift(-1, ws2812.SHIFT_CIRCULAR, 0,5)
  182. ok(equalsBuffer(buffer1, buffer2), "shift left out of bound")
  183. initBuffer(buffer1,7,8,9,12)
  184. initBuffer(buffer2,12,7,8,9)
  185. buffer1:shift(1, ws2812.SHIFT_CIRCULAR, 0,5)
  186. ok(equalsBuffer(buffer1, buffer2), "shift right out of bound")
  187. initBuffer(buffer1,7,8,9,12)
  188. initBuffer(buffer2,12,7,8,9)
  189. buffer1:shift(1, ws2812.SHIFT_CIRCULAR, -12,12)
  190. ok(equalsBuffer(buffer1, buffer2), "shift right way out of bound")
  191. end)
  192. N.test('sub', function()
  193. buffer1 = ws2812.newBuffer(4, 4)
  194. buffer2 = ws2812.newBuffer(4, 4)
  195. initBuffer(buffer1,7,8,9,12)
  196. buffer1 = buffer1:sub(4,3)
  197. ok(eq(buffer1:size(), 0), "sub empty")
  198. buffer1 = ws2812.newBuffer(4, 4)
  199. buffer2 = ws2812.newBuffer(2, 4)
  200. initBuffer(buffer1,7,8,9,12)
  201. initBuffer(buffer2,9,12)
  202. buffer1 = buffer1:sub(3,4)
  203. ok(equalsBuffer(buffer1, buffer2), "sub")
  204. buffer1 = ws2812.newBuffer(4, 4)
  205. buffer2 = ws2812.newBuffer(4, 4)
  206. initBuffer(buffer1,7,8,9,12)
  207. initBuffer(buffer2,7,8,9,12)
  208. buffer1 = buffer1:sub(-12,33)
  209. ok(equalsBuffer(buffer1, buffer2), "out of bounds")
  210. end)
  211. --[[
  212. ws2812.buffer:__concat()
  213. --]]