Add tests for setting an element in a 4d ndarray

kgryte · kgryte · commit f2d7f1f62987 · 2017-10-30T18:58:53.000-03:00
diff --git a/lib/node_modules/@stdlib/types/ndarray/ctor/test/test.instance.set.js b/lib/node_modules/@stdlib/types/ndarray/ctor/test/test.instance.set.js
@@ -802,3 +802,81 @@ tape( 'an ndarray constructor returns an instance which has a `set` method for s
 
 	t.end();
 });
+
+tape( 'an ndarray constructor returns an instance which has a `set` method for setting an array element using subscripts (4d; row-major)', function test( t ) {
+	var strides;
+	var buffer;
+	var offset;
+	var dtype;
+	var order;
+	var shape;
+	var arr;
+	var f;
+
+	dtype = 'generic';
+	buffer = [ 1.0, 2.0, 3.0, 4.0 ];
+	shape = [ 1, 1, 2, 2 ];
+	order = 'row-major';
+	strides = [ 4, 4, 2, 1 ];
+	offset = 0;
+
+	f = ctor( dtype, shape.length );
+	arr = f( buffer, shape, strides, offset, order );
+
+	t.strictEqual( hasOwnProp( arr, 'set' ), false, 'does not have own property' );
+	t.strictEqual( hasProp( arr, 'set' ), true, 'has property' );
+	t.strictEqual( isFunction( arr.set ), true, 'has method' );
+
+	arr.set( 0, 0, 0, 0, 5.0 );
+	arr.set( 0, 0, 0, 1, 6.0 );
+	arr.set( 0, 0, 1, 0, 7.0 );
+	arr.set( 0, 0, 1, 1, 8.0 );
+
+	t.strictEqual( arr.get( 0, 0, 0, 0 ), 5.0, 'returns expected value' );
+	t.strictEqual( arr.get( 0, 0, 0, 1 ), 6.0, 'returns expected value' );
+	t.strictEqual( arr.get( 0, 0, 1, 0 ), 7.0, 'returns expected value' );
+	t.strictEqual( arr.get( 0, 0, 1, 1 ), 8.0, 'returns expected value' );
+
+	t.deepEqual( buffer, [ 5.0, 6.0, 7.0, 8.0 ], 'has expected values' );
+
+	t.end();
+});
+
+tape( 'an ndarray constructor returns an instance which has a `set` method for setting an array element using subscripts (4d; column-major)', function test( t ) {
+	var strides;
+	var buffer;
+	var offset;
+	var dtype;
+	var order;
+	var shape;
+	var arr;
+	var f;
+
+	dtype = 'generic';
+	buffer = [ 1.0, 2.0, 3.0, 4.0 ];
+	shape = [ 1, 1, 2, 2 ];
+	order = 'column-major';
+	strides = [ 1, 1, 1, 2 ];
+	offset = 0;
+
+	f = ctor( dtype, shape.length );
+	arr = f( buffer, shape, strides, offset, order );
+
+	t.strictEqual( hasOwnProp( arr, 'set' ), false, 'does not have own property' );
+	t.strictEqual( hasProp( arr, 'set' ), true, 'has property' );
+	t.strictEqual( isFunction( arr.set ), true, 'has method' );
+
+	arr.set( 0, 0, 0, 0, 5.0 );
+	arr.set( 0, 0, 0, 1, 6.0 );
+	arr.set( 0, 0, 1, 0, 7.0 );
+	arr.set( 0, 0, 1, 1, 8.0 );
+
+	t.strictEqual( arr.get( 0, 0, 0, 0 ), 5.0, 'returns expected value' );
+	t.strictEqual( arr.get( 0, 0, 0, 1 ), 6.0, 'returns expected value' );
+	t.strictEqual( arr.get( 0, 0, 1, 0 ), 7.0, 'returns expected value' );
+	t.strictEqual( arr.get( 0, 0, 1, 1 ), 8.0, 'returns expected value' );
+
+	t.deepEqual( buffer, [ 5.0, 7.0, 6.0, 8.0 ], 'has expected values' );
+
+	t.end();
+});
