fix test

fluencelabs · May 31, 2024 · 1e30e6f · 1e30e6f
1 parent 97ecb13
commit 1e30e6f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/integration-tests/src/__test__/examples.spec.ts b/integration-tests/src/__test__/examples.spec.ts
@@ -877,7 +877,7 @@ describe("Testing examples", () => {
     const [resEmpty, resFirst, resSecond] = await testKeysFuncCall();
     expect(resEmpty).toEqual([]);
     expect(resFirst).toEqual(["key one"]);
-    expect(resSecond).toEqual(["key one", "key two"]);
+    expect(resSecond).toEqual(["key one", "key two", "key one", "key one"]);
   });
 
   it("streamMap.aqua keysStream function call", async () => {

diff --git a/model/inline/src/main/scala/aqua/model/inline/raw/ApplyStreamMapRawInliner.scala b/model/inline/src/main/scala/aqua/model/inline/raw/ApplyStreamMapRawInliner.scala
@@ -157,7 +157,7 @@ object ApplyStreamMapRawInliner {
       CanonicalizeModel(mapVar, CallModel.Export(canonMap.name, canonMap.`type`)).leaf,
       idxModel,
       FlattenModel(
-        canonMap.withProperty(IntoIndexModel(idx.name, CanonStreamType(mapType.element))),
+        canonMap.withProperty(IntoIndexModel(idx.name, ArrayType(mapType.element))),
         resultName
       ).leaf
     )