first and last should return nil when empty

GarrisonJ · Apr 24, 2024 · 8f8d3c8 · 8f8d3c8
1 parent 693c67e
commit 8f8d3c8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lib/sorted_containers/sorted_hash.rb b/lib/sorted_containers/sorted_hash.rb
@@ -58,6 +58,7 @@ def delete(key)
     #
     # @return [Array] A two-element array containing the key and value of the first key-value pair.
     def first
+      return nil if @sorted_array.empty?
       key = @sorted_array.first
       [key, @hash[key]]
     end
@@ -66,6 +67,7 @@ def first
     #
     # @return [Array] A two-element array containing the key and value of the first key-value pair.
     def last
+      return nil if @sorted_array.empty?
       key = @sorted_array.last
       [key, @hash[key]]
     end