Update Set#merge for Ruby 3.3 behaviour

Support multiple arguments
natalie-lang · Sep 24, 2024 · 4a3c4ac · 4a3c4ac
1 parent f7b8a73
commit 4a3c4ac
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/lib/set.rb b/lib/set.rb
@@ -288,11 +288,13 @@ def each(&block)
     end
   end
 
-  def merge(other)
-    unless other.is_a?(Enumerable)
-      raise ArgumentError, "value must be enumerable"
+  def merge(*others)
+    others.each do |other|
+      unless other.is_a?(Enumerable)
+        raise ArgumentError, "value must be enumerable"
+      end
+      other.each { |element| add(element) }
     end
-    other.each { |element| add(element) }
     self
   end