Add support for providing a dtype enum (and other dtype values)

Author: kgryte

lib/node_modules/@stdlib/ndarray/next-dtype/docs/repl.txt

@@ -11,8 +11,8 @@
 
     Parameters
     ----------
-    dtype: string (optional)
-        ndarray data type.
+    dtype: any (optional)
+        ndarray data type value.
 
     Returns
     -------

lib/node_modules/@stdlib/ndarray/next-dtype/docs/types/index.d.ts

@@ -27,14 +27,14 @@
 * -   If a data type does not have a next larger data type or the next larger type is not supported, the function returns `-1`.
 * -   If provided an unrecognized data type, the function returns `null`.
 *
-* @param dtype - ndarray data type
+* @param dtype - ndarray data type value
 * @returns next larger data type(s) or null
 *
 * @example
 * var dt = nextDataType( 'float32' );
 * // returns 'float64'
 */
-declare function nextDataType( dtype?: string ): any;
+declare function nextDataType( dtype?: any ): any;
 
 
 // EXPORTS //

lib/node_modules/@stdlib/ndarray/next-dtype/docs/types/test.ts

@@ -28,16 +28,6 @@ import nextDataType = require( './index' );
 	nextDataType( 'float' ); // $ExpectType any
 }
 
-// The function does not compile if provided a value other than a string...
-{
-	nextDataType( 123 ); // $ExpectError
-	nextDataType( true ); // $ExpectError
-	nextDataType( false ); // $ExpectError
-	nextDataType( null ); // $ExpectError
-	nextDataType( {} ); // $ExpectError
-	nextDataType( ( x: number ): number => x ); // $ExpectError
-}
-
 // The function does not compile if provided more than one argument...
 {
 	nextDataType( 'float32', 123 ); // $ExpectError

lib/node_modules/@stdlib/ndarray/next-dtype/lib/main.js

@@ -22,6 +22,7 @@
 
 var objectKeys = require( '@stdlib/utils/keys' );
 var hasOwnProp = require( '@stdlib/assert/has-own-property' );
+var resolveStr = require( '@stdlib/ndarray/base/dtype-resolve-str' );
 var NEXT_DTYPES = require( './next_dtypes.json' );
 
 
@@ -54,7 +55,7 @@ function generateTable() {
 /**
 * Returns the next larger ndarray data type of the same kind.
 *
-* @param {string} [dtype] - ndarray data type
+* @param {*} [dtype] - ndarray data type value
 * @returns {(Object|string|integer|null)} next larger data type(s) or null
 *
 * @example
@@ -65,6 +66,7 @@ function nextDataType( dtype ) {
 	if ( arguments.length === 0 ) {
 		return generateTable();
 	}
+	dtype = resolveStr( dtype );
 	if ( hasOwnProp( NEXT_DTYPES, dtype ) ) {
 		return NEXT_DTYPES[ dtype ];
 	}