Fix high water mark bug when providing an array and add tests

kgryte · kgryte · commit 421a918a7e52 · 2018-06-30T12:14:42.000-07:00
diff --git a/lib/node_modules/@stdlib/array/pool/lib/factory.js b/lib/node_modules/@stdlib/array/pool/lib/factory.js
@@ -203,6 +203,9 @@ function factory( options ) {
 			arr = arguments[ 0 ];
 			len = arr.length;
 			out = typedarray( ctor, len, dtype );
+			if ( out === null ) {
+				return out;
+			}
 			for ( i = 0; i < len; i++ ) {
 				out[ i ] = arr[ i ];
 			}
diff --git a/lib/node_modules/@stdlib/array/pool/test/test.factory.js b/lib/node_modules/@stdlib/array/pool/test/test.factory.js
@@ -1799,6 +1799,15 @@ tape( 'the function supports specifying a high water mark which limits the total
 	arr2 = typedarraypool.calloc( 1e6, 'float64' );
 	t.strictEqual( arr2, null, 'returns expected value' );
 
+	arr2 = typedarraypool( new Float64Array( 20 ), 'float64' );
+	t.strictEqual( arr2, null, 'returns expected value' );
+
+	arr2 = typedarraypool.malloc( new Float64Array( 20 ), 'float64' );
+	t.strictEqual( arr2, null, 'returns expected value' );
+
+	arr2 = typedarraypool.calloc( new Float64Array( 20 ), 'float64' );
+	t.strictEqual( arr2, null, 'returns expected value' );
+
 	typedarraypool.free( arr1 );
 
 	arr2 = typedarraypool( 1e6, 'float64' );
@@ -1810,6 +1819,15 @@ tape( 'the function supports specifying a high water mark which limits the total
 	arr2 = typedarraypool.calloc( 1e6, 'float64' );
 	t.strictEqual( arr2, null, 'returns expected value' );
 
+	arr2 = typedarraypool( new Float64Array( 20 ), 'float64' );
+	t.strictEqual( arr2, null, 'returns expected value' );
+
+	arr2 = typedarraypool.malloc( new Float64Array( 20 ), 'float64' );
+	t.strictEqual( arr2, null, 'returns expected value' );
+
+	arr2 = typedarraypool.calloc( new Float64Array( 20 ), 'float64' );
+	t.strictEqual( arr2, null, 'returns expected value' );
+
 	arr2 = typedarraypool( 8, 'float64' );
 	t.strictEqual( arr2.length, 8, 'returns expected value' );
 	t.strictEqual( arr2.byteLength, 64, 'returns expected value' );
