From a17bc04d159ec9839cc8cfb02dc0cdd2802110f4 Mon Sep 17 00:00:00 2001 From: Benoit Daloze Date: Sun, 29 Sep 2019 18:01:32 +0200 Subject: Update to ruby/spec@e69a14c --- spec/ruby/core/enumerator/lazy/flat_map_spec.rb | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'spec/ruby/core/enumerator') diff --git a/spec/ruby/core/enumerator/lazy/flat_map_spec.rb b/spec/ruby/core/enumerator/lazy/flat_map_spec.rb index 6a3391a503..5dcaa8bfa1 100644 --- a/spec/ruby/core/enumerator/lazy/flat_map_spec.rb +++ b/spec/ruby/core/enumerator/lazy/flat_map_spec.rb @@ -5,4 +5,12 @@ require_relative 'shared/collect_concat' describe "Enumerator::Lazy#flat_map" do it_behaves_like :enumerator_lazy_collect_concat, :flat_map + + it "properly unwraps nested yields" do + s = Enumerator.new do |y| loop do y << [1, 2] end end + + expected = s.take(3).flat_map { |x| x }.to_a + actual = s.lazy.take(3).flat_map{ |x| x }.force + actual.should == expected + end end -- cgit v1.2.3