Replace function head with List.wrap/1

This commit is contained in:
Jay Hayes
2016-10-17 07:54:12 -05:00
parent 9ee83336d6
commit 146da99c32

View File

@@ -1,7 +1,7 @@
defmodule Blanks do
def replace(ast, replacement) when not is_list(replacement), do: replace(ast, [replacement])
def replace([do: ast], replacements), do: [do: replace(ast, replacements)]
def replace(ast, replacements) do
replacements = List.wrap(replacements)
ast
|> Macro.prewalk(replacements, &pre/2)
|> elem(0)