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

const { assert, test, module: describe } = require('qunit');
const { CPUFunctionNode, WebGLFunctionNode, WebGL2FunctionNode } = require('../../src');

describe('internal: function node');
///
/// Test the various basic functionality of functionNode
///

/// Test the creation of a hello_world function
test("hello_world: just return magic 42 cpu", () => {
	// Create a function hello node
	const node = new CPUFunctionNode(
		(function() {
			return 42;
		}).toString(), { name: 'hello_world', output: [1] }
	);

	assert.notEqual(node.getJsAST(), null, "AST fetch check");

	assert.equal(
		node.toString().replace(/\s+/g,' '),
		"function hello_world() { return 42; }",
		"function conversion check"
	);
});

test("hello_world: just return magic 42 webgl", () => {
	// Create a function hello node
	const node = new WebGLFunctionNode(
		(function() {
			return 42;
		}).toString(), { name: 'hello_world', output: [1] }
	);

	assert.notEqual(node.getJsAST(), null, "AST fetch check");

	assert.equal(
		node.toString().replace(/\s+/g,' '),
		"float hello_world() { return 42.0; }",
		"function conversion check"
	);
});

test("hello_world: just return magic 42 webgl2", () => {
	// Create a function hello node
	const node = new WebGL2FunctionNode(
		(function() {
			return 42;
		}).toString(), { name: 'hello_world', output: [1] }
	);

	assert.notEqual(node.getJsAST(), null, "AST fetch check");

	assert.equal(
		node.toString().replace(/\s+/g,' '),
		"float hello_world() { return 42.0; }",
		"function conversion check"
	);
});

/// Test creation of function, that calls another function
test("hello_inner: call a function inside a function cpu", () => {
	function inner() {
		return 42;
	}

	// Create a function hello node
	const node = new CPUFunctionNode(
		(function() {
			return inner();
		}).toString(), { name: 'hello_inner', output: [1] }
	);

	assert.notEqual(node.getJsAST(), null, "AST fetch check");

	assert.equal(
		node.toString().replace(/\s+/g,' '),
		"function hello_inner() { return inner(); }",
		"function conversion check"
	);

	assert.deepEqual(node.calledFunctions, ["inner"] );
});

test("hello_inner: call a function inside a function webgl", () => {
	function inner() {
		return 42;
	}

	// Create a function hello node
	const node = new WebGLFunctionNode(
		(function() {
			return inner();
		}).toString(), { name: 'hello_inner', output: [1] }
	);

	assert.notEqual(node.getJsAST(), null, "AST fetch check");

	assert.equal(
		node.toString().replace(/\s+/g,' '),
		"float hello_inner() { return inner(); }",
		"function conversion check"
	);

	assert.deepEqual(node.calledFunctions, ["inner"] );
});

/// Test creation of function, that calls another function
test("hello_inner: call a function inside a function webgl2", () => {
	function inner() {
		return 42;
	}

	// Create a function hello node
	const node = new WebGL2FunctionNode(
		(function() {
			return inner();
		}).toString(), { name: 'hello_inner', output: [1] }
	);

	assert.notEqual(node.getJsAST(), null, "AST fetch check");

	assert.equal(
		node.toString().replace(/\s+/g,' '),
		"float hello_inner() { return inner(); }",
		"function conversion check"
	);

	assert.deepEqual(node.calledFunctions, ["inner"] );
});

/// Test creation of function, that calls another function, with ARGS
test("Math.round implementation: A function with arguments cpu", () => {
	// Math.round node
	const node = new CPUFunctionNode(
		(function(a) {
			return Math.floor(a + 0.5 );
		}).toString(), { name: 'foo', output: [1] }
	);

	assert.notEqual(node.getJsAST(), null, "AST fetch check");

	assert.equal(
		node.toString().replace(/\s+/g,' ').replace(/user_/g,''),
		"function foo(a) { return Math.floor((a+0.5)); }",
		"function conversion check"
	);

	assert.deepEqual(node.calledFunctions, ["Math.floor"]);
});

test("Math.round implementation: A function with arguments webgl", () => {
	// Math.round node
	const node = new WebGLFunctionNode(
		(function(a) {
			return Math.floor(a + 0.5 );
		}).toString(), { name: 'foo', output: [1] }
	);

	assert.notEqual(node.getJsAST(), null, "AST fetch check");

	assert.equal(
		node.toString().replace(/\s+/g,' ').replace(/user_/g,''),
		"float foo(float a) { return floor((a+0.5)); }",
		"function conversion check"
	);

	assert.deepEqual(node.calledFunctions, ["floor"] );
});

test("Math.round implementation: A function with arguments webgl2", () => {
	// Math.round node
	const node = new WebGL2FunctionNode(
		(function(a) {
			return Math.floor(a + 0.5 );
		}).toString(), { name: 'foo', output: [1] }
	);

	assert.notEqual(node.getJsAST(), null, "AST fetch check");

	assert.equal(
		node.toString().replace(/\s+/g,' ').replace(/user_/g,''),
		"float foo(float a) { return floor((a+0.5)); }",
		"function conversion check"
	);

	assert.deepEqual(node.calledFunctions, ["floor"] );
});

/// Test creation of function, that calls another function, with ARGS
test("Two arguments test webgl", function(assert){
	const node = new WebGLFunctionNode(
		(function(a,b) {
			return a+b;
		}).toString(), { name: 'add_together', output: [1] }
	);

	assert.notEqual(node.getJsAST(), null, "AST fetch check");

	assert.equal(
		node.toString().replace(/\s+/g,' ').replace(/user_/g,''),
		"float add_together(float a, float b) { return (a+b); }",
		"function conversion check"
	);
});

test("Two arguments test webgl2", function(assert){
	const node = new WebGL2FunctionNode(
		(function(a,b) {
			return a+b;
		}).toString(), { name: 'add_together', output: [1] }
	);

	assert.notEqual(node.getJsAST(), null, "AST fetch check");

	assert.equal(
		node.toString().replace(/\s+/g,' ').replace(/user_/g,''),
		"float add_together(float a, float b) { return (a+b); }",
		"function conversion check"
	);
});

/// Test the creation of a hello_world function
test("Automatic naming support cpu", () => {
	function hello_world() {
		return 42;
	}
	// Create a function hello node
	const node = new CPUFunctionNode(hello_world.toString(), { output: [1] });
	assert.notEqual(node, null, "class creation check");
	assert.equal(node.name, "hello_world");
});

test("Automatic naming support webgl", () => {
	function hello_world() {
		return 42;
	}
	// Create a function hello node
	const node = new WebGLFunctionNode(hello_world.toString(), { output: [1] });
	assert.notEqual(node, null, "class creation check");
	assert.equal(node.name, "hello_world");
});

test("Automatic naming support webgl2", () => {
	function hello_world() {
		return 42;
	}
	// Create a function hello node
	const node = new WebGL2FunctionNode(hello_world.toString(), { output: [1] });
	assert.notEqual(node, null, "class creation check");
	assert.equal(node.name, "hello_world");
});