fix: Add more unit tests for function return, covering Array(2), Array(3), and Array(4)

Author: robertleeplummerjr

test/features/function-return.js

@@ -1,40 +1,165 @@
-const { assert, skip, test, module: describe } = require('qunit');
+const { assert, skip, test, module: describe, only } = require('qunit');
 const { GPU } = require('../../src');
 
-describe('function return');
+describe('features: function return');
 
-function functionReturn( mode ) {
+function functionReturnFloat( mode ) {
   const gpu = new GPU({ mode });
   const f = gpu.createKernel(function() {
-    return 42.0;
+    return 42;
   }, {
     output : [1]
   });
-  assert.ok( f !== null, "function generated test");
-  assert.equal(f()[0], 42.0, "basic return function test");
+  assert.ok( f !== null, 'function generated test');
+  assert.equal(f()[0], 42);
   gpu.destroy();
 }
 
-test("auto", () => {
-  functionReturn(null);
+test('float auto', () => {
+  functionReturnFloat(null);
 });
 
-test("gpu", () => {
-  functionReturn("gpu");
+test('float gpu', () => {
+  functionReturnFloat('gpu');
 });
 
-(GPU.isWebGLSupported ? test : skip)("webgl", () => {
-  functionReturn("webgl");
+(GPU.isWebGLSupported ? test : skip)('float webgl', () => {
+  functionReturnFloat('webgl');
 });
 
-(GPU.isWebGL2Supported ? test : skip)("webgl2", () => {
-  functionReturn("webgl2");
+(GPU.isWebGL2Supported ? test : skip)('float webgl2', () => {
+  functionReturnFloat('webgl2');
 });
 
-(GPU.isHeadlessGLSupported ? test : skip)("headlessgl", () => {
-  functionReturn("headlessgl");
+(GPU.isHeadlessGLSupported ? test : skip)('float headlessgl', () => {
+  functionReturnFloat('headlessgl');
 });
 
-test("cpu", () => {
-  functionReturn("cpu");
+test('float cpu', () => {
+  functionReturnFloat('cpu');
+});
+
+
+function functionReturnArray2( mode ) {
+  const gpu = new GPU({ mode });
+  const f = gpu.createKernel(function() {
+    return [42, 43];
+  }, {
+    output : [1]
+  });
+  assert.ok( f !== null, 'function generated test');
+  const result = f();
+  // TODO, make assertion work
+  // assert(result[0].constructor, Float32Array);
+  assert.equal(result[0][0], 42);
+  assert.equal(result[0][1], 43);
+  gpu.destroy();
+}
+
+test('Array(2) auto', () => {
+  functionReturnArray2(null);
+});
+
+test('Array(2) gpu', () => {
+  functionReturnArray2('gpu');
+});
+
+(GPU.isWebGLSupported ? test : skip)('Array(2) webgl', () => {
+  functionReturnArray2('webgl');
+});
+
+(GPU.isWebGL2Supported ? test : skip)('Array(2) webgl2', () => {
+  functionReturnArray2('webgl2');
+});
+
+(GPU.isHeadlessGLSupported ? test : skip)('Array(2) headlessgl', () => {
+  functionReturnArray2('headlessgl');
+});
+
+test('Array(2) cpu', () => {
+  functionReturnArray2('cpu');
+});
+
+function functionReturnArray3( mode ) {
+  const gpu = new GPU({ mode });
+  const f = gpu.createKernel(function() {
+    return [42, 43, 44];
+  }, {
+    output : [1]
+  });
+  assert.ok( f !== null, 'function generated test');
+  const result = f();
+  // TODO, make assertion work
+  // assert(result[0].constructor, Float32Array);
+  assert.equal(result[0][0], 42);
+  assert.equal(result[0][1], 43);
+  assert.equal(result[0][2], 44);
+  gpu.destroy();
+}
+
+test('Array(3) auto', () => {
+  functionReturnArray3(null);
+});
+
+test('Array(3) gpu', () => {
+  functionReturnArray3('gpu');
+});
+
+(GPU.isWebGLSupported ? test : skip)('Array(3) webgl', () => {
+  functionReturnArray3('webgl');
+});
+
+(GPU.isWebGL2Supported ? test : skip)('Array(3) webgl2', () => {
+  functionReturnArray3('webgl2');
+});
+
+(GPU.isHeadlessGLSupported ? test : skip)('Array(3) headlessgl', () => {
+  functionReturnArray3('headlessgl');
+});
+
+test('Array(3) cpu', () => {
+  functionReturnArray3('cpu');
+});
+
+
+function functionReturnArray4( mode ) {
+  const gpu = new GPU({ mode });
+  const f = gpu.createKernel(function() {
+    return [42, 43, 44, 45];
+  }, {
+    output : [1]
+  });
+  assert.ok( f !== null, 'function generated test');
+  const result = f();
+  // TODO, make assertion work
+  // assert(result[0].constructor, Float32Array);
+  assert.equal(result[0][0], 42);
+  assert.equal(result[0][1], 43);
+  assert.equal(result[0][2], 44);
+  assert.equal(result[0][3], 45);
+  gpu.destroy();
+}
+
+test('Array(4) auto', () => {
+  functionReturnArray4(null);
+});
+
+test('Array(4) gpu', () => {
+  functionReturnArray4('gpu');
+});
+
+(GPU.isWebGLSupported ? test : skip)('Array(4) webgl', () => {
+  functionReturnArray4('webgl');
+});
+
+(GPU.isWebGL2Supported ? test : skip)('Array(4) webgl2', () => {
+  functionReturnArray4('webgl2');
+});
+
+(GPU.isHeadlessGLSupported ? test : skip)('Array(4) headlessgl', () => {
+  functionReturnArray4('headlessgl');
+});
+
+test('Array(4) cpu', () => {
+  functionReturnArray4('cpu');
 });