From 4686a0b5db232ca4d9140ee3ce34f30bbce2c6bc Mon Sep 17 00:00:00 2001
From: keiravillekode <keiravillekode@gmail.com>
Date: Mon, 4 Nov 2024 19:02:50 +1100
Subject: [PATCH] Fix flatten-array compile error (#186)

---
 exercises/practice/flatten-array/test/src/Main.idr | 6 +++---
 generators/exercises/flatten_array.py              | 2 ++
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/exercises/practice/flatten-array/test/src/Main.idr b/exercises/practice/flatten-array/test/src/Main.idr
index de53f01..d1d1217 100644
--- a/exercises/practice/flatten-array/test/src/Main.idr
+++ b/exercises/practice/flatten-array/test/src/Main.idr
@@ -8,9 +8,9 @@ import FlattenArray
 
 tests : List Test
 tests =
-  [ test "empty"                                                                               (assertEq (flatten $ Branch []) [])
+  [ test "empty"                                                                               (assertEq (flatten $ Branch []) (the (List Int) []))
   , test "no nesting"                                                                          (assertEq (flatten $ Branch [Leaf 0, Leaf 1, Leaf 2]) [0, 1, 2])
-  , test "flattens a nested array"                                                             (assertEq (flatten $ Branch [Branch [Branch []]]) [])
+  , test "flattens a nested array"                                                             (assertEq (flatten $ Branch [Branch [Branch []]]) (the (List Int) []))
   , test "flattens array with just integers present"                                           (assertEq (flatten $ Branch [Leaf 1, Branch [Leaf 2, Leaf 3, Leaf 4, Leaf 5, Leaf 6, Leaf 7], Leaf 8]) [1, 2, 3, 4, 5, 6, 7, 8])
   , test "5 level nesting"                                                                     (assertEq (flatten $ Branch [Leaf 0, Leaf 2, Branch [Branch [Leaf 2, Leaf 3], Leaf 8, Leaf 100, Leaf 4, Branch [Branch [Branch [Leaf 50]]]], Leaf (-2)]) [0, 2, 2, 3, 8, 100, 4, 50, -2])
   , test "6 level nesting"                                                                     (assertEq (flatten $ Branch [Leaf 1, Branch [Leaf 2, Branch [Branch [Leaf 3]], Branch [Leaf 4, Branch [Branch [Leaf 5]]], Leaf 6, Leaf 7], Leaf 8]) [1, 2, 3, 4, 5, 6, 7, 8])
@@ -18,7 +18,7 @@ tests =
   , test "consecutive null values at the front of the list are omitted from the final result"  (assertEq (flatten $ Branch [Hollow, Hollow, Leaf 3]) [3])
   , test "consecutive null values in the middle of the list are omitted from the final result" (assertEq (flatten $ Branch [Leaf 1, Hollow, Hollow, Leaf 4]) [1, 4])
   , test "6 level nest list with null values"                                                  (assertEq (flatten $ Branch [Leaf 0, Leaf 2, Branch [Branch [Leaf 2, Leaf 3], Leaf 8, Branch [Branch [Leaf 100]], Hollow, Branch [Branch [Hollow]]], Leaf (-2)]) [0, 2, 2, 3, 8, 100, -2])
-  , test "all values in nested list are null"                                                  (assertEq (flatten $ Branch [Hollow, Branch [Branch [Branch [Hollow]]], Hollow, Hollow, Branch [Branch [Hollow, Hollow], Hollow], Hollow]) [])
+  , test "all values in nested list are null"                                                  (assertEq (flatten $ Branch [Hollow, Branch [Branch [Branch [Hollow]]], Hollow, Hollow, Branch [Branch [Hollow, Hollow], Hollow], Hollow]) (the (List Int) []))
   ]
 
 export
diff --git a/generators/exercises/flatten_array.py b/generators/exercises/flatten_array.py
index f953441..3a21a04 100644
--- a/generators/exercises/flatten_array.py
+++ b/generators/exercises/flatten_array.py
@@ -10,5 +10,7 @@ def to_tree(data):
         return f"Leaf {data}"
     property = case["property"]
     expected = case["expected"]
+    if expected == []:
+        expected = "(the (List Int) [])"
     tree = to_tree(case["input"]["array"])
     return f'assertEq ({property} $ {tree}) {expected}'