mispec_ws2812.lua 5.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153
  1. require 'mispec'
  2. local buffer, buffer1, buffer2
  3. local function initBuffer(buffer, ...)
  4. local i,v
  5. for i,v in ipairs({...}) do
  6. buffer:set(i, v, v*2, v*3, v*4)
  7. end
  8. return buffer
  9. end
  10. local function equalsBuffer(buffer1, buffer2)
  11. return eq(buffer1:dump(), buffer2:dump())
  12. end
  13. describe('WS2812 buffers', function(it)
  14. it:should('initialize a buffer', function()
  15. buffer = ws2812.newBuffer(9, 3)
  16. ko(buffer == nil)
  17. ok(eq(buffer:size(), 9), "check size")
  18. 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")
  19. failwith("should be a positive integer", ws2812.newBuffer, 9, 0)
  20. failwith("should be a positive integer", ws2812.newBuffer, 9, -1)
  21. failwith("should be a positive integer", ws2812.newBuffer, 0, 3)
  22. failwith("should be a positive integer", ws2812.newBuffer, -1, 3)
  23. end)
  24. it:should('have correct size', function()
  25. buffer = ws2812.newBuffer(9, 3)
  26. ok(eq(buffer:size(), 9), "check size")
  27. buffer = ws2812.newBuffer(9, 22)
  28. ok(eq(buffer:size(), 9), "check size")
  29. buffer = ws2812.newBuffer(13, 1)
  30. ok(eq(buffer:size(), 13), "check size")
  31. end)
  32. it:should('fill a buffer with one color', function()
  33. buffer = ws2812.newBuffer(3, 3)
  34. buffer:fill(1,222,55)
  35. ok(eq(buffer:dump(), string.char(1,222,55,1,222,55,1,222,55)), "RGB")
  36. buffer = ws2812.newBuffer(3, 4)
  37. buffer:fill(1,222,55, 77)
  38. ok(eq(buffer:dump(), string.char(1,222,55,77,1,222,55,77,1,222,55,77)), "RGBW")
  39. end)
  40. it:should('replace correctly', function()
  41. buffer = ws2812.newBuffer(5, 3)
  42. buffer:replace(string.char(3,255,165,33,0,244,12,87,255))
  43. ok(eq(buffer:dump(), string.char(3,255,165,33,0,244,12,87,255,0,0,0,0,0,0)), "RGBW")
  44. buffer = ws2812.newBuffer(5, 3)
  45. buffer:replace(string.char(3,255,165,33,0,244,12,87,255), 2)
  46. ok(eq(buffer:dump(), string.char(0,0,0,3,255,165,33,0,244,12,87,255,0,0,0)), "RGBW")
  47. buffer = ws2812.newBuffer(5, 3)
  48. buffer:replace(string.char(3,255,165,33,0,244,12,87,255), -5)
  49. ok(eq(buffer:dump(), string.char(3,255,165,33,0,244,12,87,255,0,0,0,0,0,0)), "RGBW")
  50. failwith("Does not fit into destination", function() buffer:replace(string.char(3,255,165,33,0,244,12,87,255), 4) end)
  51. end)
  52. it:should('replace correctly issue #2921', function()
  53. local buffer = ws2812.newBuffer(5, 3)
  54. buffer:replace(string.char(3,255,165,33,0,244,12,87,255), -7)
  55. ok(eq(buffer:dump(), string.char(3,255,165,33,0,244,12,87,255,0,0,0,0,0,0)), "RGBW")
  56. end)
  57. it:should('get/set correctly', function()
  58. buffer = ws2812.newBuffer(3, 4)
  59. buffer:fill(1,222,55,13)
  60. ok(eq({buffer:get(2)},{1,222,55,13}))
  61. buffer:set(2, 4,53,99,0)
  62. ok(eq({buffer:get(1)},{1,222,55,13}))
  63. ok(eq({buffer:get(2)},{4,53,99,0}))
  64. ok(eq(buffer:dump(), string.char(1,222,55,13,4,53,99,0,1,222,55,13)), "RGBW")
  65. failwith("index out of range", function() buffer:get(0) end)
  66. failwith("index out of range", function() buffer:get(4) end)
  67. failwith("index out of range", function() buffer:set(0,1,2,3,4) end)
  68. failwith("index out of range", function() buffer:set(4,1,2,3,4) end)
  69. failwith("number expected, got no value", function() buffer:set(2,1,2,3) end)
  70. -- failwith("extra values given", function() buffer:set(2,1,2,3,4,5) end)
  71. end)
  72. it:should('fade correctly', function()
  73. buffer = ws2812.newBuffer(1, 3)
  74. buffer:fill(1,222,55)
  75. buffer:fade(2)
  76. ok(buffer:dump() == string.char(0,111,27), "RGB")
  77. buffer:fill(1,222,55)
  78. buffer:fade(3, ws2812.FADE_OUT)
  79. ok(buffer:dump() == string.char(0,222/3,55/3), "RGB")
  80. buffer:fill(1,222,55)
  81. buffer:fade(3, ws2812.FADE_IN)
  82. ok(buffer:dump() == string.char(3,255,165), "RGB")
  83. buffer = ws2812.newBuffer(1, 4)
  84. buffer:fill(1,222,55, 77)
  85. buffer:fade(2, ws2812.FADE_OUT)
  86. ok(eq(buffer:dump(), string.char(0,111,27,38)), "RGBW")
  87. end)
  88. it:should('mix correctly issue #1736', function()
  89. buffer1 = ws2812.newBuffer(1, 3)
  90. buffer2 = ws2812.newBuffer(1, 3)
  91. buffer1:fill(10,22,54)
  92. buffer2:fill(10,27,55)
  93. buffer1:mix(256/8*7,buffer1,256/8,buffer2)
  94. ok(eq({buffer1:get(1)}, {10,23,54}))
  95. end)
  96. it:should('mix saturation correctly ', function()
  97. buffer1 = ws2812.newBuffer(1, 3)
  98. buffer2 = ws2812.newBuffer(1, 3)
  99. buffer1:fill(10,22,54)
  100. buffer2:fill(10,27,55)
  101. buffer1:mix(256/2,buffer1,-256,buffer2)
  102. ok(eq({buffer1:get(1)}, {0,0,0}))
  103. buffer1:fill(10,22,54)
  104. buffer2:fill(10,27,55)
  105. buffer1:mix(25600,buffer1,256/8,buffer2)
  106. ok(eq({buffer1:get(1)}, {255,255,255}))
  107. buffer1:fill(10,22,54)
  108. buffer2:fill(10,27,55)
  109. buffer1:mix(-257,buffer1,255,buffer2)
  110. ok(eq({buffer1:get(1)}, {0,5,1}))
  111. end)
  112. it:should('mix with strings correctly ', function()
  113. buffer1 = ws2812.newBuffer(1, 3)
  114. buffer2 = ws2812.newBuffer(1, 3)
  115. buffer1:fill(10,22,54)
  116. buffer2:fill(10,27,55)
  117. buffer1:mix(-257,buffer1:dump(),255,buffer2:dump())
  118. ok(eq({buffer1:get(1)}, {0,5,1}))
  119. end)
  120. it:should('power', function()
  121. buffer = ws2812.newBuffer(2, 4)
  122. buffer:fill(10,22,54,234)
  123. ok(eq(buffer:power(), 2*(10+22+54+234)))
  124. end)
  125. end)
  126. mispec.run()