flatten.ex

defmodule Flatten do

  def flatten([]), do: []
  def flatten([head | tail]), do: flatten(head) ++ flatten(tail)
  def flatten(x), do: [x]

end

defmodule FlattenTest do
  use ExUnit.Case
  doctest Flatten

  test "preserves structure" do
    assert [1, 2, 3] == Flatten.flatten([1, 2, 3])
  end

  test "flattens a shallow nested list" do
    assert [1,2,3,4] == Flatten.flatten([1,2,[3],4])
  end

  test "flattens a deeply nested list" do
    assert [1,2,3,4] == Flatten.flatten([[1,2,[3]],4])
  end
end