Further improve quantileSeq typings

josdejong · Jun 21, 2024 · 94e3f35 · 94e3f35
1 parent 5785cb9
commit 94e3f35
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/test/typescript-tests/testTypes.ts b/test/typescript-tests/testTypes.ts
@@ -2543,6 +2543,12 @@ Statistics functions' return types
   >()
 
   expectTypeOf(math.quantileSeq([1, 2, 3], 0.75)).toMatchTypeOf<number>()
+  expectTypeOf(math.quantileSeq([1, 2, 3, 4, 5], [0.25, 0.75])).toMatchTypeOf<
+    MathArray | MathScalarType
+  >()
+  expectTypeOf(
+    math.quantileSeq([1, 2, 3, 4, 5], [0.25, 0.75]) as number[]
+  ).toMatchTypeOf<number[]>()
   expectTypeOf(math.quantileSeq([[1, 2, 3]], 0.75)).toMatchTypeOf<number>()
   expectTypeOf(
     math.quantileSeq([math.bignumber('123')], 0.75)

diff --git a/types/index.d.ts b/types/index.d.ts
@@ -2866,7 +2866,7 @@ export interface MathJsInstance extends MathJsFactory {
    */
   quantileSeq<T extends MathScalarType>(
     A: T[] | T[][],
-    prob: number | BigNumber | MathArray,
+    prob: number | BigNumber,
     sorted?: boolean
   ): T
   /**