Update Enumerable#each_cons and Enumerable#each_slice

ruby/ruby#1509
ruby · Feb 1, 2022 · 530cf57 · 530cf57
1 parent 7bb18e6
commit 530cf57
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 4 deletions.
diff --git a/core/enumerable.rbs b/core/enumerable.rbs
@@ -484,8 +484,8 @@ module Enumerable[unchecked out Elem] : _Each[Elem]
   #
   # With no block given, returns an Enumerator.
   #
-  def each_cons: (Integer n) { (::Array[Elem] arg0) -> untyped } -> NilClass
-               | (Integer n) -> ::Enumerator[::Array[Elem], NilClass]
+  def each_cons: (Integer n) { (::Array[Elem]) -> void } -> self
+               | (Integer n) -> ::Enumerator[::Array[Elem], self]
 
   # <!--
   #   rdoc-file=enum.c
@@ -1501,8 +1501,8 @@ module Enumerable[unchecked out Elem] : _Each[Elem]
   #
   # With no block given, returns an Enumerator.
   #
-  def each_slice: (Integer n) { (::Array[Elem]) -> untyped } -> NilClass
-                | (Integer n) -> ::Enumerator[::Array[Elem], NilClass]
+  def each_slice: (Integer n) { (::Array[Elem]) -> void } -> self
+                | (Integer n) -> ::Enumerator[::Array[Elem], self]
 
   interface _NotFound[T]
     def call: () -> T

diff --git a/test/stdlib/Enumerable_test.rb b/test/stdlib/Enumerable_test.rb
@@ -180,6 +180,20 @@ def test_each_with_object
                      TestEnumerable.new, :each_with_object, 0 do end
   end
 
+  def test_each_cons
+    assert_send_type(
+      "(Integer) { (Array[String]) -> void } -> EnumerableTest2::TestEnumerable",
+      TestEnumerable.new, :each_cons, 2
+    ) do end
+  end
+
+  def test_each_slice
+    assert_send_type(
+      "(Integer) { (Array[String]) -> void } -> EnumerableTest2::TestEnumerable",
+      TestEnumerable.new, :each_slice, 2
+    ) do end
+  end
+
   def test_find_index
     assert_send_type "() -> ::Enumerator[String, Integer?]", TestEnumerable.new,
                      :find_index