Solve 'Matrix creation' kata

Author: borisskert

src/MatrixCreation.hs

@@ -0,0 +1,8 @@
+module MatrixCreation (getMatrix) where
+
+-- https://www.codewars.com/kata/5a34da5dee1aae516d00004a/train/haskell
+
+getMatrix :: Int -> [[Int]]
+getMatrix x = map makeRow [0 .. x - 1]
+  where
+    makeRow n = replicate n 0 ++ [1] ++ replicate (x - n - 1) 0

test/MatrixCreationSpec.hs

@@ -0,0 +1,26 @@
+module MatrixCreationSpec (spec) where
+
+import MatrixCreation (getMatrix)
+import Test.HUnit (assertEqual)
+import Test.Hspec
+
+spec :: Spec
+spec = do
+  it "example tests" $ do
+    assertEqual "getMatrix 0" [] $ getMatrix 0
+    assertEqual "getMatrix 1" [[1]] $ getMatrix 1
+    assertEqual
+      "getMatrix 2"
+      [ [1, 0],
+        [0, 1]
+      ]
+      $ getMatrix 2
+    assertEqual
+      "getMatrix 5"
+      [ [1, 0, 0, 0, 0],
+        [0, 1, 0, 0, 0],
+        [0, 0, 1, 0, 0],
+        [0, 0, 0, 1, 0],
+        [0, 0, 0, 0, 1]
+      ]
+      $ getMatrix 5