Day 8

test/Day6Spec.hs

@@ -2,7 +2,6 @@ module Day6Spec (spec) where
 
 import Test.Hspec
 import Day6
-import qualified Data.Sequence as S
 
 spec :: Spec
 spec =

test/Day8Spec.hs

@@ -0,0 +1,65 @@
+{-# LANGUAGE QuasiQuotes #-}
+module Day8Spec (spec) where
+
+import Test.Hspec
+import Text.Heredoc
+import Control.Arrow ((>>>))
+
+import qualified Data.Matrix as M
+import qualified Data.Vector as V
+import Day8
+
+testInput :: String
+testInput = [str|30373
+                |25512
+                |65332
+                |33549
+                |35390
+                |]
+
+testVisibility :: Visibillity
+testVisibility = M.fromList 5 5
+  [x,x,x,x,x,
+   x,x,x,o,x,
+   x,x,o,x,x,
+   x,o,x,o,x,
+   x,x,x,x,x
+  ]
+  where
+    x = True
+    o = False
+
+smallForest :: String
+smallForest = [str|1234
+                  |5678
+                  |9012
+                  |]
+
+parsedSmallForest :: Forest
+parsedSmallForest = M.fromList 3 4 $ [1..9] ++ [0,1,2]
+
+spec :: Spec
+spec =
+  describe "Day8" $ do
+    describe "Part1" $ do
+      it "parses a forest" $ do
+        parseForest smallForest `shouldBe` parsedSmallForest
+      it "checks visibility" $ do
+        let forest = parseForest testInput
+        checkVisibility forest `shouldBe` testVisibility
+      it "counts visible things" $ do
+        let v = parseForest >>>
+                checkVisibility >>>
+                countVisible $ testInput
+        v `shouldBe` 21
+    describe "Part2" $ do
+      it "calculates viewing distance from height" $ do
+        viewingDistance 5 (V.fromList [3]) `shouldBe` 1
+        viewingDistance 5 (V.fromList [5,2]) `shouldBe` 1
+        viewingDistance 5 (V.fromList [1,2]) `shouldBe` 2
+        viewingDistance 5 (V.fromList [3,5,3]) `shouldBe` 2
+      it "calculates the scenic score" $ do
+        let s = parseForest >>>
+                findMaxScenicScore $ testInput
+        s `shouldBe` 8
+