312 lines
7.8 KiB
Haskell
312 lines
7.8 KiB
Haskell
{-# LANGUAGE OverloadedStrings #-}
|
|
module Main (main) where
|
|
|
|
import Test.HUnit (Test(..), assertEqual, runTestTT, failures)
|
|
import qualified System.Exit as Exit
|
|
import qualified Day01
|
|
import qualified Day02
|
|
import qualified Day03
|
|
import qualified Day04
|
|
import qualified Day05
|
|
import qualified Day06
|
|
import qualified Day07
|
|
import qualified Day08
|
|
import qualified Day09
|
|
import qualified Day10
|
|
import qualified Day15
|
|
|
|
|
|
day01 :: Test
|
|
day01 =
|
|
let parsedA = Day01.parse inputA
|
|
parsedB = Day01.parse inputB
|
|
in TestList
|
|
[ TestCase $ assertEqual "A" (Right 142) (Day01.solveA <$> parsedA)
|
|
, TestCase $ assertEqual "B" (Right 281) (Day01.solveB <$> parsedB)
|
|
]
|
|
where
|
|
inputA =
|
|
"1abc2\n\
|
|
\pqr3stu8vwx\n\
|
|
\a1b2c3d4e5f\n\
|
|
\treb7uchet\n"
|
|
inputB =
|
|
"two1nine\n\
|
|
\eightwothree\n\
|
|
\abcone2threexyz\n\
|
|
\xtwone3four\n\
|
|
\4nineeightseven2\n\
|
|
\zoneight234\n\
|
|
\7pqrstsixteen\n"
|
|
|
|
day02 :: Test
|
|
day02 =
|
|
let parsed = Day02.parse input
|
|
in TestList
|
|
[ TestCase $ assertEqual "A" (Right 8) (Day02.solveA <$> parsed)
|
|
, TestCase $ assertEqual "B" (Right 2286) (Day02.solveB <$> parsed)
|
|
]
|
|
where
|
|
input =
|
|
"Game 1: 3 blue, 4 red; 1 red, 2 green, 6 blue; 2 green\n\
|
|
\Game 2: 1 blue, 2 green; 3 green, 4 blue, 1 red; 1 green, 1 blue\n\
|
|
\Game 3: 8 green, 6 blue, 20 red; 5 blue, 4 red, 13 green; 5 green, 1 red\n\
|
|
\Game 4: 1 green, 3 red, 6 blue; 3 green, 6 red; 3 green, 15 blue, 14 red\n\
|
|
\Game 5: 6 red, 1 blue, 3 green; 2 blue, 1 red, 2 green\n"
|
|
|
|
day03 :: Test
|
|
day03 =
|
|
let parsed = Day03.parse input
|
|
in TestList
|
|
[ TestCase $ assertEqual "A" (Right 4361) (Day03.solveA <$> parsed)
|
|
, TestCase $ assertEqual "B" (Right 467835) (Day03.solveB <$> parsed)
|
|
]
|
|
where
|
|
input =
|
|
"467..114..\n\
|
|
\...*......\n\
|
|
\..35..633.\n\
|
|
\......#...\n\
|
|
\617*......\n\
|
|
\.....+.58.\n\
|
|
\..592.....\n\
|
|
\......755.\n\
|
|
\...$.*....\n\
|
|
\.664.598..\n"
|
|
|
|
day04 :: Test
|
|
day04 =
|
|
let parsed = Day04.parse input
|
|
in TestList
|
|
[ TestCase $ assertEqual "A" (Right 13) (Day04.solveA <$> parsed)
|
|
, TestCase $ assertEqual "B" (Right 30) (Day04.solveB <$> parsed)
|
|
]
|
|
where
|
|
input =
|
|
"Card 1: 41 48 83 86 17 | 83 86 6 31 17 9 48 53\n\
|
|
\Card 2: 13 32 20 16 61 | 61 30 68 82 17 32 24 19\n\
|
|
\Card 3: 1 21 53 59 44 | 69 82 63 72 16 21 14 1\n\
|
|
\Card 4: 41 92 73 84 69 | 59 84 76 51 58 5 54 83\n\
|
|
\Card 5: 87 83 26 28 32 | 88 30 70 12 93 22 82 36\n\
|
|
\Card 6: 31 18 13 56 72 | 74 77 10 23 35 67 36 11\n"
|
|
|
|
day05 :: Test
|
|
day05 =
|
|
let parsed = Day05.parse input
|
|
in TestList
|
|
[ TestCase $ assertEqual "A" (Right 35) (Day05.solveA <$> parsed)
|
|
, TestCase $ assertEqual "B" (Right 46) (Day05.solveB <$> parsed)
|
|
]
|
|
where
|
|
input =
|
|
"seeds: 79 14 55 13\n\
|
|
\\n\
|
|
\seed-to-soil map:\n\
|
|
\50 98 2\n\
|
|
\52 50 48\n\
|
|
\\n\
|
|
\soil-to-fertilizer map:\n\
|
|
\0 15 37\n\
|
|
\37 52 2\n\
|
|
\39 0 15\n\
|
|
\\n\
|
|
\fertilizer-to-water map:\n\
|
|
\49 53 8\n\
|
|
\0 11 42\n\
|
|
\42 0 7\n\
|
|
\57 7 4\n\
|
|
\\n\
|
|
\water-to-light map:\n\
|
|
\88 18 7\n\
|
|
\18 25 70\n\
|
|
\\n\
|
|
\light-to-temperature map:\n\
|
|
\45 77 23\n\
|
|
\81 45 19\n\
|
|
\68 64 13\n\
|
|
\\n\
|
|
\temperature-to-humidity map:\n\
|
|
\0 69 1\n\
|
|
\1 0 69\n\
|
|
\\n\
|
|
\humidity-to-location map:\n\
|
|
\60 56 37\n\
|
|
\56 93 4\n"
|
|
|
|
day06 :: Test
|
|
day06 =
|
|
let parsed = Day06.parse input
|
|
in TestList
|
|
[ TestCase $ assertEqual "A" (Right 288) (Day06.solveA <$> parsed)
|
|
, TestCase $ assertEqual "B" (Right 71503) (Day06.solveB <$> parsed)
|
|
]
|
|
where
|
|
input =
|
|
"Time: 7 15 30\n\
|
|
\Distance: 9 40 200\n"
|
|
|
|
day07 :: Test
|
|
day07 =
|
|
let parsed = Day07.parse input
|
|
in TestList
|
|
[ TestCase $ assertEqual "A" (Right 6440) (Day07.solveA <$> parsed)
|
|
, TestCase $ assertEqual "B" (Right 5905) (Day07.solveB <$> parsed)
|
|
]
|
|
where
|
|
input =
|
|
"32T3K 765\n\
|
|
\T55J5 684\n\
|
|
\KK677 28\n\
|
|
\KTJJT 220\n\
|
|
\QQQJA 483\n"
|
|
|
|
day08 :: Test
|
|
day08 =
|
|
let parsedA1 = Day08.parse inputA1
|
|
parsedA2 = Day08.parse inputA2
|
|
parsedB = Day08.parse inputB
|
|
in TestList
|
|
[ TestCase $ assertEqual "A" (Right 2) (Day08.solveA <$> parsedA1)
|
|
, TestCase $ assertEqual "A" (Right 6) (Day08.solveA <$> parsedA2)
|
|
, TestCase $ assertEqual "B" (Right 6) (Day08.solveB <$> parsedB)
|
|
]
|
|
where
|
|
inputA1 =
|
|
"RL\n\
|
|
\\n\
|
|
\AAA = (BBB, CCC)\n\
|
|
\BBB = (DDD, EEE)\n\
|
|
\CCC = (ZZZ, GGG)\n\
|
|
\DDD = (DDD, DDD)\n\
|
|
\EEE = (EEE, EEE)\n\
|
|
\GGG = (GGG, GGG)\n\
|
|
\ZZZ = (ZZZ, ZZZ)\n"
|
|
inputA2 =
|
|
"LLR\n\
|
|
\\n\
|
|
\AAA = (BBB, BBB)\n\
|
|
\BBB = (AAA, ZZZ)\n\
|
|
\ZZZ = (ZZZ, ZZZ)\n"
|
|
inputB =
|
|
"LR\n\
|
|
\\n\
|
|
\11A = (11B, XXX)\n\
|
|
\11B = (XXX, 11Z)\n\
|
|
\11Z = (11B, XXX)\n\
|
|
\22A = (22B, XXX)\n\
|
|
\22B = (22C, 22C)\n\
|
|
\22C = (22Z, 22Z)\n\
|
|
\22Z = (22B, 22B)\n\
|
|
\XXX = (XXX, XXX)\n"
|
|
|
|
day09 :: Test
|
|
day09 =
|
|
let parsed = Day09.parse input
|
|
in TestList
|
|
[ TestCase $ assertEqual "A" (Right 114) (Day09.solveA <$> parsed)
|
|
, TestCase $ assertEqual "B" (Right 2) (Day09.solveB <$> parsed)
|
|
]
|
|
where
|
|
input =
|
|
"0 3 6 9 12 15\n\
|
|
\1 3 6 10 15 21\n\
|
|
\10 13 16 21 30 45\n"
|
|
|
|
day10 :: Test
|
|
day10 =
|
|
let parsedA = Day10.parse inputA
|
|
parsedB1 = Day10.parse inputB1
|
|
parsedB2 = Day10.parse inputB2
|
|
parsedB3 = Day10.parse inputB3
|
|
parsedB4 = Day10.parse inputB4
|
|
in TestList
|
|
[ TestCase $ assertEqual "A" (Right 4) (Day10.solveA <$> parsedA)
|
|
, TestCase $ assertEqual "B" (Right 4) (Day10.solveB <$> parsedB1)
|
|
, TestCase $ assertEqual "B" (Right 4) (Day10.solveB <$> parsedB2)
|
|
, TestCase $ assertEqual "B" (Right 8) (Day10.solveB <$> parsedB3)
|
|
, TestCase $ assertEqual "B" (Right 10) (Day10.solveB <$> parsedB4)
|
|
]
|
|
where
|
|
inputA =
|
|
".....\n\
|
|
\.S-7.\n\
|
|
\.|.|.\n\
|
|
\.L-J.\n\
|
|
\.....\n"
|
|
inputB1 =
|
|
"...........\n\
|
|
\.S-------7.\n\
|
|
\.|F-----7|.\n\
|
|
\.||.....||.\n\
|
|
\.||.....||.\n\
|
|
\.|L-7.F-J|.\n\
|
|
\.|..|.|..|.\n\
|
|
\.L--J.L--J.\n\
|
|
\...........\n"
|
|
inputB2 =
|
|
"..........\n\
|
|
\.S------7.\n\
|
|
\.|F----7|.\n\
|
|
\.||OOOO||.\n\
|
|
\.||OOOO||.\n\
|
|
\.|L-7F-J|.\n\
|
|
\.|II||II|.\n\
|
|
\.L--JL--J.\n\
|
|
\..........\n"
|
|
inputB3 =
|
|
".F----7F7F7F7F-7....\n\
|
|
\.|F--7||||||||FJ....\n\
|
|
\.||.FJ||||||||L7....\n\
|
|
\FJL7L7LJLJ||LJ.L-7..\n\
|
|
\L--J.L7...LJS7F-7L7.\n\
|
|
\....F-J..F7FJ|L7L7L7\n\
|
|
\....L7.F7||L7|.L7L7|\n\
|
|
\.....|FJLJ|FJ|F7|.LJ\n\
|
|
\....FJL-7.||.||||...\n\
|
|
\....L---J.LJ.LJLJ...\n"
|
|
inputB4 =
|
|
"FF7FSF7F7F7F7F7F---7\n\
|
|
\L|LJ||||||||||||F--J\n\
|
|
\FL-7LJLJ||||||LJL-77\n\
|
|
\F--JF--7||LJLJ7F7FJ-\n\
|
|
\L---JF-JLJ.||-FJLJJ7\n\
|
|
\|F|F-JF---7F7-L7L|7|\n\
|
|
\|FFJF7L7F-JF7|JL---7\n\
|
|
\7-L-JL7||F7|L7F-7F7|\n\
|
|
\L.L7LFJ|||||FJL7||LJ\n\
|
|
\L7JLJL-JLJLJL--JLJ.L\n"
|
|
|
|
day15 :: Test
|
|
day15 =
|
|
let parsedA1 = Day15.parse inputA1
|
|
parsedA2 = Day15.parse inputA2
|
|
in TestList
|
|
[ TestCase $ assertEqual "A" (Right 52) (Day15.solveA <$> parsedA1)
|
|
, TestCase $ assertEqual "A" (Right 1320) (Day15.solveA <$> parsedA2)
|
|
]
|
|
where
|
|
inputA1 = "HASH"
|
|
inputA2 = "rn=1,cm-,qp=3,cm=2,qp-,pc=4,ot=9,ab=5,pc-,pc=6,ot=7"
|
|
|
|
tests :: Test
|
|
tests = TestList
|
|
[ TestLabel "01" day01
|
|
, TestLabel "02" day02
|
|
, TestLabel "03" day03
|
|
, TestLabel "04" day04
|
|
, TestLabel "05" day05
|
|
, TestLabel "06" day06
|
|
, TestLabel "07" day07
|
|
, TestLabel "08" day08
|
|
, TestLabel "09" day09
|
|
, TestLabel "10" day10
|
|
, TestLabel "15" day15
|
|
]
|
|
|
|
main :: IO ()
|
|
main = isFailed . failures =<< runTestTT tests
|
|
where
|
|
isFailed count
|
|
| count > 0 = Exit.exitFailure
|
|
| otherwise = Exit.exitSuccess
|