Flatten arrays

require 'minitest/autorun'
class ArrayTest < Minitest::Test
  def test_simple_array
    assert [[1,2,[3]],4].flattify == [1,2,3,4]
    assert [1,2,[3],4].flattify == [1,2,3,4]
    assert [[[1],[2]],3,[4]].flattify == [1,2,3,4]
    assert [['a'],4].flattify == ['a',4]
  end


end

class Array
  #iterate through every element, recurse if it the element is array
  def flattify
    puts self.is_a? Array
    each_with_object([]) do |element, flatten_array|
      flatten_array.push *(element.is_a?(Array) ? element.flattify : element)
    end
  rescue StandardError => e
    #log e
  end
end