Add tests

Author: zacklukem

2024/lib/day21.ex

@@ -95,7 +95,11 @@ defmodule Day21a do
   end
 
   def run do
-    File.read!(".input.txt")
+    File.read!(".input.txt") |> run
+  end
+
+  def run(input) do
+    input
     |> String.split("\n")
     |> Enum.map(fn code ->
       state = ~c"AAA"
@@ -242,7 +246,11 @@ defmodule Day21b do
   end
 
   def run do
-    File.read!(".input.txt")
+    File.read!(".input.txt") |> run
+  end
+
+  def run(input) do
+    input
     |> String.split("\n")
     |> Enum.map(fn code ->
       state = 0..25 |> Enum.map(fn _ -> ?A end)

2024/test/day21_test.exs

@@ -0,0 +1,17 @@
+defmodule Day21Test do
+  use ExUnit.Case
+
+  def example, do: "029A
+980A
+179A
+456A
+379A"
+
+  test "Part 1 Example" do
+    assert Day21a.run(example()) == 126_384
+  end
+
+  test "Part 2 Example" do
+    assert Day21b.run(example()) == 154_115_708_116_294
+  end
+end

2024/test/test_helper.exs

@@ -0,0 +1 @@
+ExUnit.start()