gpu.js/test/internal/function-builder.js

var GPU = require('../../src/index');

(function() {
  ///
  /// Test the various basic functionality of functionBuilder
  ///

  // Three layer template for multiple tests
  function threeLayerTemplate() {
    function layerOne() {
      return 42;
    }

    function layerTwo() {
      return layerOne() * 2;
    }

    function layerThree() {
      return layerTwo() * 2;
    }

    // Create a function hello node
    var builder = new GPU.WebGLFunctionBuilder();
    builder.addFunction(null, layerOne);
    builder.addFunction(null, layerTwo);
    builder.addFunction(null, layerThree);
    return builder;
  }

  /// Test the function tracing of 3 layers
  QUnit.test( 'traceFunctionCalls: 3 layer test', function( assert ) {
    assert.notEqual( GPU.WebGLFunctionBuilder, null, 'script include check' );
    var builder = threeLayerTemplate();
    assert.notEqual( builder, null, 'class creation check' );

    assert.deepEqual( builder.traceFunctionCalls('layerOne'),   ['layerOne'] );
    assert.deepEqual( builder.traceFunctionCalls('layerTwo'),   ['layerTwo','layerOne'] );
    assert.deepEqual( builder.traceFunctionCalls('layerThree'), ['layerThree','layerTwo','layerOne'] );
  });

  /// Test the function tracing of 3 layers
  QUnit.test( 'webglString: 3 layer test', function( assert ) {
    assert.notEqual( GPU.WebGLFunctionBuilder, null, 'script include check' );
    var builder = threeLayerTemplate();
    assert.notEqual( builder, null, 'class creation check' );

    assert.equal(
      builder.getStringFromFunctionNames(['layerOne']),
      'float layerOne() {\nreturn 42.0;\n}'
    );
    assert.equal(
      builder.getString('layerOne'),
      builder.getStringFromFunctionNames(['layerOne'])
    );

    assert.equal(
      builder.getStringFromFunctionNames(['layerOne','layerTwo']),
      'float layerOne() {\nreturn 42.0;\n}\nfloat layerTwo() {\nreturn (layerOne()*2.0);\n}'
    );
    assert.equal(
      builder.getString('layerTwo'),
      builder.getStringFromFunctionNames(['layerOne','layerTwo'])
    );


    assert.equal(
      builder.getStringFromFunctionNames(['layerOne','layerTwo','layerThree']),
      'float layerOne() {\nreturn 42.0;\n}\nfloat layerTwo() {\nreturn (layerOne()*2.0);\n}\nfloat layerThree() {\nreturn (layerTwo()*2.0);\n}'
    );
    assert.equal(
      builder.getString('layerThree'),
      builder.getStringFromFunctionNames(['layerOne','layerTwo','layerThree'])
    );
    assert.equal(
      builder.getString(null),
      builder.getString('layerThree')
    );
  });
})();