Add tests for setting elements in a 3d ndarray

kgryte · kgryte · commit 4493d002411c · 2017-10-30T18:09:28.000-03:00
diff --git a/lib/node_modules/@stdlib/types/ndarray/ctor/test/test.instance.set_no_codegen.js b/lib/node_modules/@stdlib/types/ndarray/ctor/test/test.instance.set_no_codegen.js
@@ -375,8 +375,8 @@ tape( 'an ndarray constructor returns an instance which has a `set` method for s
 		'codegen': false
 	};
 
-	dtype = 'float64';
-	buffer = new Float64Array( [ 1.0, 2.0, 3.0, 4.0 ] );
+	dtype = 'generic';
+	buffer = [ 1.0, 2.0, 3.0, 4.0 ];
 	shape = [ 4 ];
 	order = 'row-major';
 	strides = [ 1 ];
@@ -419,8 +419,8 @@ tape( 'an ndarray constructor returns an instance which has a `set` method for s
 		'codegen': false
 	};
 
-	dtype = 'float64';
-	buffer = new Float64Array( [ 1.0, 2.0, 3.0, 4.0 ] );
+	dtype = 'generic';
+	buffer = [ 1.0, 2.0, 3.0, 4.0 ];
 	shape = [ 4 ];
 	order = 'column-major';
 	strides = [ 1 ];
@@ -799,3 +799,91 @@ 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 (3d; row-major; codegen=false)', function test( t ) {
+	var strides;
+	var buffer;
+	var offset;
+	var dtype;
+	var order;
+	var shape;
+	var opts;
+	var arr;
+	var f;
+
+	opts = {
+		'codegen': false
+	};
+
+	dtype = 'generic';
+	buffer = [ 1.0, 2.0, 3.0, 4.0 ];
+	shape = [ 1, 2, 2 ];
+	order = 'row-major';
+	strides = [ 4, 2, 1 ];
+	offset = 0;
+
+	f = ctor( dtype, shape.length, opts );
+	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, 5.0 );
+	arr.set( 0, 0, 1, 6.0 );
+	arr.set( 0, 1, 0, 7.0 );
+	arr.set( 0, 1, 1, 8.0 );
+
+	t.strictEqual( arr.get( 0, 0, 0 ), 5.0, 'returns expected value' );
+	t.strictEqual( arr.get( 0, 0, 1 ), 6.0, 'returns expected value' );
+	t.strictEqual( arr.get( 0, 1, 0 ), 7.0, 'returns expected value' );
+	t.strictEqual( arr.get( 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 (3d; column-major; codegen=false)', function test( t ) {
+	var strides;
+	var buffer;
+	var offset;
+	var dtype;
+	var order;
+	var shape;
+	var opts;
+	var arr;
+	var f;
+
+	opts = {
+		'codegen': false
+	};
+
+	dtype = 'generic';
+	buffer = [ 1.0, 2.0, 3.0, 4.0 ];
+	shape = [ 1, 2, 2 ];
+	order = 'column-major';
+	strides = [ 1, 1, 2 ];
+	offset = 0;
+
+	f = ctor( dtype, shape.length, opts );
+	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, 5.0 );
+	arr.set( 0, 0, 1, 6.0 );
+	arr.set( 0, 1, 0, 7.0 );
+	arr.set( 0, 1, 1, 8.0 );
+
+	t.strictEqual( arr.get( 0, 0, 0 ), 5.0, 'returns expected value' );
+	t.strictEqual( arr.get( 0, 0, 1 ), 6.0, 'returns expected value' );
+	t.strictEqual( arr.get( 0, 1, 0 ), 7.0, 'returns expected value' );
+	t.strictEqual( arr.get( 0, 1, 1 ), 8.0, 'returns expected value' );
+
+	t.deepEqual( buffer, [ 5.0, 7.0, 6.0, 8.0 ], 'has expected values' );
+
+	t.end();
+});
