convert function name to string in remover

lib/ex_factor/remover.ex

@@ -9,8 +9,11 @@ defmodule ExFactor.Remover do
   Remove the indicated function and its spec from it's original file.
   """
   def remove(opts) do
-    source_function = Keyword.fetch!(opts, :source_function)
+    source_function = opts
+    |> Keyword.fetch!(:source_function)
+    |> function_name()
     source_module = Keyword.get(opts, :source_module)
+
     arity = Keyword.fetch!(opts, :arity)
     source_path = Keyword.get(opts, :source_path, path(source_module))
     dry_run = Keyword.get(opts, :dry_run, false)
@@ -66,4 +69,9 @@ defmodule ExFactor.Remover do
   end
 
   defp path(module), do: ExFactor.path(module)
+
+  defp function_name(name) when is_binary(name) do
+    String.to_atom(name)
+  end
+  defp function_name(name), do: name
 end

test/ex_factor/remover_test.exs

@@ -42,7 +42,7 @@ defmodule ExFactor.RemoverTest do
       File.rm("test/support/source_module.ex")
     end
 
-    test "it rewrites the source file and removes code blocks" do
+    test "it rewrites the source file and removes code blocks when function is a string" do
       module = """
       defmodule ExFactorSampleModule do
         @somedoc "This is somedoc"
@@ -73,7 +73,7 @@ defmodule ExFactor.RemoverTest do
       opts = [
         source_module: ExFactorSampleModule,
         source_path: "test/tmp/source_module.ex",
-        source_function: :pub1,
+        source_function: "pub1",
         arity: 1
       ]