2 files changed, 6 insertions, 9 deletions
diff --git a/test/ruby/test_enumerator.rb b/test/ruby/test_enumerator.rb
index 5fabea645d..177d7c04fa 100644
--- a/test/ruby/test_enumerator.rb
+++ b/test/ruby/test_enumerator.rb
@@ -1070,13 +1070,4 @@ class TestEnumerator < Test::Unit::TestCase
     enum = ary.each
     assert_equal(35.0, enum.sum)
   end
-
-  def test_to_set
-    e = Enumerator.new { it << 1 << 1 << 2 << 3 << 5 }
-    set = e.to_set
-    assert_equal(Set[1, 2, 3, 5], set)
-
-    ei = Enumerator.new(Float::INFINITY) { it << 1 << 1 << 2 << 3 << 5 }
-    assert_raise(ArgumentError) { ei.to_set }
-  end
 end
diff --git a/test/ruby/test_set.rb b/test/ruby/test_set.rb
index e8ac3e329e..70a61aa3b5 100644
--- a/test/ruby/test_set.rb
+++ b/test/ruby/test_set.rb
@@ -1003,6 +1003,12 @@ class TC_Enumerable < Test::Unit::TestCase
     set = assert_nothing_raised { enum.to_set }
     assert(set.is_a?(Set))
     assert_equal(Set[1,2,3], set)
+
+    enumerator = enum.to_enum
+
+    set = assert_nothing_raised { enumerator.to_set }
+    assert(set.is_a?(Set))
+    assert_equal(Set[1,2,3], set)
   end
 end