Skip to content

Commit

Permalink
Implements #flatten! method
Browse files Browse the repository at this point in the history
  • Loading branch information
hslzr committed Nov 28, 2024
1 parent 1936498 commit 4d25e9f
Show file tree
Hide file tree
Showing 2 changed files with 16 additions and 0 deletions.
4 changes: 4 additions & 0 deletions lib/literal/array.rb
Original file line number Diff line number Diff line change
Expand Up @@ -243,6 +243,10 @@ def first(...)
@__value__.first(...)
end

def flatten!(...)
@__value__.flatten!(...) ? self : nil
end

def freeze
@__value__.freeze
super
Expand Down
12 changes: 12 additions & 0 deletions test/array.test.rb
Original file line number Diff line number Diff line change
Expand Up @@ -472,3 +472,15 @@
array.narrow(Numeric)
}.to_raise(ArgumentError)
end

test "#flatten! flattens the array" do
array = Literal::Array(Array).new([1, 2], [3, 4])

expect((array.flatten!).to_a) == [1, 2, 3, 4]
end

test "#flatten! returns nil if no nested arrays" do
array = Literal::Array(Integer).new(1, 2, 3)

expect(array.flatten!) == nil
end

0 comments on commit 4d25e9f

Please sign in to comment.