test: add argument tests

Author: kgryte

lib/node_modules/@stdlib/ndarray/map/test/test.js

@@ -607,3 +607,129 @@ tape( 'the function supports providing a callback execution context (options)',
 		return z * 10.0;
 	}
 });
+
+tape( 'the function invokes a provided callback with three arguments (row-major)', function test( t ) {
+	var expected;
+	var indices;
+	var values;
+	var arrays;
+	var ord;
+	var sh;
+	var st;
+	var dt;
+	var o;
+	var x;
+	var y;
+	var i;
+
+	dt = 'float64';
+	ord = 'row-major';
+	sh = [ 2, 1, 2 ];
+	st = shape2strides( sh, ord );
+	o = strides2offset( sh, st );
+
+	x = ndarray( dt, ones( numel( sh ), dt ), sh, st, o, ord );
+
+	values = [];
+	indices = [];
+	arrays = [];
+	y = map( x, scale );
+
+	expected = new Float64Array([
+		10.0,
+		10.0,
+		10.0,
+		10.0
+	]);
+	t.strictEqual( isSameFloat64Array( y.data, expected ), true, 'returns expected value' );
+
+	expected = [
+		[ 0, 0, 0 ],
+		[ 0, 0, 1 ],
+		[ 1, 0, 0 ],
+		[ 1, 0, 1 ]
+	];
+	t.deepEqual( indices, expected, 'returns expected value' );
+
+	expected = [
+		x,
+		x,
+		x,
+		x
+	];
+	for ( i = 0; i < expected.length; i++ ) {
+		t.strictEqual( arrays[ i ], expected[ i ], 'returns expected value' );
+	}
+
+	t.end();
+
+	function scale( z, idx, arr ) {
+		values.push( z );
+		indices.push( idx );
+		arrays.push( arr );
+		return z * 10.0;
+	}
+});
+
+tape( 'the function invokes a provided callback with three arguments (column-major)', function test( t ) {
+	var expected;
+	var indices;
+	var values;
+	var arrays;
+	var ord;
+	var sh;
+	var st;
+	var dt;
+	var o;
+	var x;
+	var y;
+	var i;
+
+	dt = 'float64';
+	ord = 'column-major';
+	sh = [ 2, 1, 2 ];
+	st = shape2strides( sh, ord );
+	o = strides2offset( sh, st );
+
+	x = ndarray( dt, ones( numel( sh ), dt ), sh, st, o, ord );
+
+	values = [];
+	indices = [];
+	arrays = [];
+	y = map( x, scale );
+
+	expected = new Float64Array([
+		10.0,
+		10.0,
+		10.0,
+		10.0
+	]);
+	t.strictEqual( isSameFloat64Array( y.data, expected ), true, 'returns expected value' );
+
+	expected = [
+		[ 0, 0, 0 ],
+		[ 1, 0, 0 ],
+		[ 0, 0, 1 ],
+		[ 1, 0, 1 ]
+	];
+	t.deepEqual( indices, expected, 'returns expected value' );
+
+	expected = [
+		x,
+		x,
+		x,
+		x
+	];
+	for ( i = 0; i < expected.length; i++ ) {
+		t.strictEqual( arrays[ i ], expected[ i ], 'returns expected value' );
+	}
+
+	t.end();
+
+	function scale( z, idx, arr ) {
+		values.push( z );
+		indices.push( idx );
+		arrays.push( arr );
+		return z * 10.0;
+	}
+});