ruby iteration

iterate.rb

test1 = "1\n2\n3"
test2 = [1, 2, 3]

def iterate (obj)
  method = obj.is_a?(Array) ? :each : :each_line

  obj.send(method) do |line|
    yield line
  end
end

iterate (test1) do |line|
  p line
end
# => 1\n, 2\n, 3\n
iterate (test2) do |line|
  p line
end
# => 1, 2, 3