feat: add boolean dtype support to array/base/count-truthy

kgryte · kgryte · commit 09f1e11557f7 · 2024-06-19T17:35:13.000-07:00
Ref: stdlib-js#2304
diff --git a/lib/node_modules/@stdlib/array/base/count-truthy/lib/main.js b/lib/node_modules/@stdlib/array/base/count-truthy/lib/main.js
@@ -21,9 +21,11 @@
 // MODULES //
 
 var isComplexTypedArray = require( '@stdlib/array/base/assert/is-complex-typed-array' );
+var isBooleanArray = require( '@stdlib/array/base/assert/is-booleanarray' );
 var isAccessorArray = require( '@stdlib/array/base/assert/is-accessor-array' );
 var resolveGetter = require( '@stdlib/array/base/resolve-getter' );
-var reinterpret = require( '@stdlib/strided/base/reinterpret-complex' );
+var reinterpretComplex = require( '@stdlib/strided/base/reinterpret-complex' );
+var reinterpretBoolean = require( '@stdlib/strided/base/reinterpret-boolean' );
 
 
 // FUNCTIONS //
@@ -105,7 +107,7 @@ function complex( x ) {
 	var n;
 	var i;
 
-	view = reinterpret( x, 0 );
+	view = reinterpretComplex( x, 0 );
 
 	n = 0;
 	for ( i = 0; i < view.length; i += 2 ) {
@@ -133,6 +135,9 @@ function complex( x ) {
 */
 function countTruthy( x ) {
 	if ( isAccessorArray( x ) ) {
+		if ( isBooleanArray( x ) ) {
+			return indexed( reinterpretBoolean( x, 0 ) );
+		}
 		if ( isComplexTypedArray( x ) ) {
 			return complex( x );
 		}
diff --git a/lib/node_modules/@stdlib/array/base/count-truthy/test/test.js b/lib/node_modules/@stdlib/array/base/count-truthy/test/test.js
@@ -22,6 +22,7 @@
 
 var tape = require( 'tape' );
 var Complex128Array = require( '@stdlib/array/complex128' );
+var BooleanArray = require( '@stdlib/array/bool' );
 var Int32Array = require( '@stdlib/array/int32' );
 var toAccessorArray = require( '@stdlib/array/base/to-accessor-array' );
 var countTruthy = require( './../lib' );
@@ -87,6 +88,19 @@ tape( 'the function counts the number of truthy values (complex typed array)', f
 	t.end();
 });
 
+tape( 'the function counts the number of truthy values (boolean typed array)', function test( t ) {
+	var expected;
+	var actual;
+	var x;
+
+	x = new BooleanArray( [ true, true, false, true ] );
+	expected = 3;
+	actual = countTruthy( x );
+
+	t.strictEqual( actual, expected, 'returns expected value' );
+	t.end();
+});
+
 tape( 'the function returns zero if provided an array of length `0`', function test( t ) {
 	var expected;
 	var actual;
@@ -110,5 +124,9 @@ tape( 'the function returns zero if provided an array of length `0`', function t
 	actual = countTruthy( x );
 	t.strictEqual( actual, expected, 'returns expected value' );
 
+	x = new BooleanArray( [] );
+	actual = countTruthy( x );
+	t.strictEqual( actual, expected, 'returns expected value' );
+
 	t.end();
 });
