diff --git a/lib/display.ex b/lib/display.ex index 1e25203..7c64c0f 100644 --- a/lib/display.ex +++ b/lib/display.ex @@ -6,7 +6,7 @@ defmodule Display do IO.puts("") IO.puts("Now meditate upon #{display_module(module)}") IO.puts("---------------------------------------") - IO.puts("Assertion failed in #{source_file(module)}:#{line_number(expr)}") + IO.puts(format_cyan(display_failed_assertion(module, expr))) IO.puts(display_koan(name)) IO.puts(format_red(Macro.to_string(expr))) end @@ -20,6 +20,10 @@ defmodule Display do IO.puts("") end + def display_failed_assertion(module, expr) do + "Assertion failed in #{source_file(module)}:#{line_number(expr)}" + end + defp line_number({_, [line: line], _}) do line end @@ -34,6 +38,10 @@ defmodule Display do Enum.join([ANSI.red, str, ANSI.reset], "") end + defp format_cyan(str) do + Enum.join([ANSI.cyan, str, ANSI.reset], "") + end + defp display_module(module) do Module.split(module) |> List.last end