add push and append to SortedArray

GarrisonJ · Apr 29, 2024 · 99ab784 · 99ab784
1 parent 19b33d2
commit 99ab784
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/lib/sorted_containers/sorted_array.rb b/lib/sorted_containers/sorted_array.rb
@@ -93,8 +93,11 @@ def add(value)
         expand(pos)
       end
       @size += 1
+      self
     end
     alias << add
+    alias push add
+    alias append add
 
     # rubocop:enable Metrics/MethodLength
 

diff --git a/spec/sorted_array_spec.rb b/spec/sorted_array_spec.rb
@@ -144,6 +144,11 @@
       expect(array.to_a).to eq((1..1000).to_a)
     end
 
+    it "should return the array" do
+      array = SortedContainers::SortedArray.new
+      expect(array.add(5)).to eq(array)
+    end
+
     it "a load factor of 10 should work" do
       array = SortedContainers::SortedArray.new([], load_factor: 10)
       (1..1000).to_a.shuffle.each do |i|