From 2ae3b6cf7b9cae3af4066f160a6537a4b044c56d Mon Sep 17 00:00:00 2001
From: bluepython508 <bluepython508@users.noreply.github.io>
Date: Sun, 3 Dec 2023 10:00:10 +0000
Subject: [PATCH] Further output improvements

---
 lib/mix_tasks.ex | 16 ++++++----------
 1 file changed, 6 insertions(+), 10 deletions(-)

diff --git a/lib/mix_tasks.ex b/lib/mix_tasks.ex
index efe53d7..add3140 100644
--- a/lib/mix_tasks.ex
+++ b/lib/mix_tasks.ex
@@ -131,16 +131,12 @@ defmodule Mix.Tasks.Aoc do
       p2 = mod.part2(parsed)
 
       result = fn (part, got, expected) ->
-        if expected do
-          if stringify(got) != expected do
-            IO.puts(IO.ANSI.format [:red, "Test #{test}.#{part} failed: expected #{expected}"])
-            dbg(got)
-          else
-            IO.puts(IO.ANSI.format [:green, "Test #{test}.#{part} succeeded"])
-          end
-        else
-          IO.puts("Test #{test}.#{part}:")
-          dbg(got)
+        cond do
+          stringify(got) == expected ->
+            IO.puts(IO.ANSI.format [:green, "Test #{test}.#{part} succeeded (#{expected})"])
+          expected ->
+            IO.puts(IO.ANSI.format [:red, "Test #{test}.#{part} failed: expected #{expected}, got #{stringify(got)}"])
+          true ->  IO.puts("Test #{test}.#{part}: #{stringify(got)}")
         end
       end