Add tests

Author: jablko

tests/baselines/reference/arrayFlatMap.js

@@ -3,10 +3,18 @@ const array: number[] = [];
 const readonlyArray: ReadonlyArray<number> = [];
 array.flatMap((): ReadonlyArray<number> => []); // ok
 readonlyArray.flatMap((): ReadonlyArray<number> => []); // ok
+
+// #19535
+
+declare function f(): number[] | number[][];
+const [x] = [1].flatMap(f);
+x == 1;
 
 
 //// [arrayFlatMap.js]
 var array = [];
 var readonlyArray = [];
 array.flatMap(function () { return []; }); // ok
 readonlyArray.flatMap(function () { return []; }); // ok
+var x = [1].flatMap(f)[0];
+x == 1;

tests/baselines/reference/arrayFlatMap.symbols

@@ -18,3 +18,17 @@ readonlyArray.flatMap((): ReadonlyArray<number> => []); // ok
 >flatMap : Symbol(ReadonlyArray.flatMap, Decl(lib.es2019.array.d.ts, --, --))
 >ReadonlyArray : Symbol(ReadonlyArray, Decl(lib.es5.d.ts, --, --), Decl(lib.es2015.core.d.ts, --, --), Decl(lib.es2015.iterable.d.ts, --, --), Decl(lib.es2016.array.include.d.ts, --, --), Decl(lib.es2019.array.d.ts, --, --))
 
+// #19535
+
+declare function f(): number[] | number[][];
+>f : Symbol(f, Decl(arrayFlatMap.ts, 3, 55))
+
+const [x] = [1].flatMap(f);
+>x : Symbol(x, Decl(arrayFlatMap.ts, 8, 7))
+>[1].flatMap : Symbol(Array.flatMap, Decl(lib.es2019.array.d.ts, --, --))
+>flatMap : Symbol(Array.flatMap, Decl(lib.es2019.array.d.ts, --, --))
+>f : Symbol(f, Decl(arrayFlatMap.ts, 3, 55))
+
+x == 1;
+>x : Symbol(x, Decl(arrayFlatMap.ts, 8, 7))
+

tests/baselines/reference/arrayFlatMap.types

@@ -23,3 +23,22 @@ readonlyArray.flatMap((): ReadonlyArray<number> => []); // ok
 >(): ReadonlyArray<number> => [] : () => readonly number[]
 >[] : undefined[]
 
+// #19535
+
+declare function f(): number[] | number[][];
+>f : () => number[] | number[][]
+
+const [x] = [1].flatMap(f);
+>x : number | number[]
+>[1].flatMap(f) : (number | number[])[]
+>[1].flatMap : <U, This = undefined, That extends number[]>(this: readonly number[], callback: (this: This, value: That[number], index: number, array: That) => U, thisArg?: This) => (U extends readonly (infer V)[] ? V : U)[]
+>[1] : number[]
+>1 : 1
+>flatMap : <U, This = undefined, That extends number[]>(this: readonly number[], callback: (this: This, value: That[number], index: number, array: That) => U, thisArg?: This) => (U extends readonly (infer V)[] ? V : U)[]
+>f : () => number[] | number[][]
+
+x == 1;
+>x == 1 : boolean
+>x : number | number[]
+>1 : 1
+

tests/cases/compiler/arrayFlatMap.ts

@@ -4,3 +4,9 @@ const array: number[] = [];
 const readonlyArray: ReadonlyArray<number> = [];
 array.flatMap((): ReadonlyArray<number> => []); // ok
 readonlyArray.flatMap((): ReadonlyArray<number> => []); // ok
+
+// #19535
+
+declare function f(): number[] | number[][];
+const [x] = [1].flatMap(f);
+x == 1;