Add support for dtype enums (and other dtype values)

Author: kgryte

lib/node_modules/@stdlib/ndarray/promotion-rules/docs/repl.txt

@@ -12,11 +12,11 @@
 
     Parameters
     ----------
-    dtype1: string (optional)
-        ndarray data type.
+    dtype1: any (optional)
+        ndarray data type value.
 
-    dtype2: string (optional)
-        ndarray data type.
+    dtype2: any (optional)
+        ndarray data type value.
 
     Returns
     -------

lib/node_modules/@stdlib/ndarray/promotion-rules/docs/types/index.d.ts

@@ -29,8 +29,8 @@ interface Table {
 /**
 * Returns the ndarray data type with the smallest size and closest "kind" to which ndarray data types can be safely cast.
 *
-* @param dtype1 - ndarray data type
-* @param dtype2 - ndarray data type
+* @param dtype1 - ndarray data type value
+* @param dtype2 - ndarray data type value
 * @returns promotion rule(s) or null
 *
 * @example
@@ -45,7 +45,7 @@ interface Table {
 * var dt = promotionRules( 'float32', 'foo' );
 * // returns null
 */
-declare function promotionRules( dtype1: string, dtype2: string ): number | string | null; // tslint-disable-line max-line-length
+declare function promotionRules( dtype1: any, dtype2: any ): number | string | null; // tslint-disable-line max-line-length
 
 /**
 * Returns a type promotion table displaying the ndarray data types with the smallest size and closest "kind" to which ndarray data types can be safely cast.

lib/node_modules/@stdlib/ndarray/promotion-rules/docs/types/test.ts

@@ -29,26 +29,6 @@ import promotionRules = require( './index' );
 	promotionRules( 'float32', 'foo' ); // $ExpectType string | number | null
 }
 
-// The function does not compile if provided a first argument that is not a string...
-{
-	promotionRules( 123, 'float64' ); // $ExpectError
-	promotionRules( true, 'float64' ); // $ExpectError
-	promotionRules( false, 'float64' ); // $ExpectError
-	promotionRules( null, 'float64' ); // $ExpectError
-	promotionRules( {}, 'float64' ); // $ExpectError
-	promotionRules( ( x: number ): number => x, 'float64' ); // $ExpectError
-}
-
-// The function does not compile if provided a second argument that is not a string...
-{
-	promotionRules( 'int32', 123 ); // $ExpectError
-	promotionRules( 'int32', true ); // $ExpectError
-	promotionRules( 'int32', false ); // $ExpectError
-	promotionRules( 'int32', null ); // $ExpectError
-	promotionRules( 'int32', {} ); // $ExpectError
-	promotionRules( 'int32', ( x: number ): number => x ); // $ExpectError
-}
-
 // The function does not compile if provided more than one argument...
 {
 	promotionRules( 'float32' ); // $ExpectError

lib/node_modules/@stdlib/ndarray/promotion-rules/lib/main.js

@@ -22,6 +22,7 @@
 
 var objectKeys = require( '@stdlib/utils/keys' );
 var hasOwnProp = require( '@stdlib/assert/has-own-property' );
+var resolve = require( '@stdlib/ndarray/base/dtype-resolve-str' );
 var PROMOTION_RULES = require( './promotion_rules.json' );
 
 
@@ -66,8 +67,8 @@ function generateFullTable() {
 /**
 * Returns the ndarray data type with the smallest size and closest "kind" to which ndarray data types can be safely cast.
 *
-* @param {string} [dtype1] - ndarray data type
-* @param {string} [dtype2] - ndarray data type
+* @param {*} [dtype1] - ndarray data type value
+* @param {*} [dtype2] - ndarray data type value
 * @returns {(Object|integer|string|null)} promotion rule(s) or null
 *
 * @example
@@ -91,8 +92,10 @@ function promotionRules( dtype1, dtype2 ) {
 	if ( arguments.length === 0 ) {
 		return generateFullTable();
 	}
+	dtype1 = resolve( dtype1 );
 	if ( hasOwnProp( PROMOTION_RULES, dtype1 ) ) {
 		o = PROMOTION_RULES[ dtype1 ];
+		dtype2 = resolve( dtype2 );
 		if ( hasOwnProp( o, dtype2 ) ) {
 			return o[ dtype2 ];
 		}