* enumerator.c: Fix size for Enumerator::Lazy#flat_map

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@38808 b2dd03c8-39d4-4d8f-98ff-823fe69b080e

ChangeLog

@@ -1,3 +1,7 @@
+Mon Jan 14 10:18:56 2013  Marc-Andre Lafortune  <ruby-core@marc-andre.ca>
+
+	* enumerator.c: Fix size for Enumerator::Lazy#flat_map
+
 Mon Jan 14 07:12:52 2013  Marc-Andre Lafortune  <ruby-core@marc-andre.ca>
 
 	* lib/matrix/lup_decomposition: Fix bugs with LUP Decomposition of

enumerator.c

@@ -1486,7 +1486,7 @@ lazy_flat_map(VALUE obj)
 
     return lazy_set_method(rb_block_call(rb_cLazy, id_new, 1, &obj,
 					 lazy_flat_map_func, 0),
-			   Qnil, lazy_receiver_size);
+			   Qnil, 0);
 }
 
 static VALUE

test/ruby/test_lazy_enumerator.rb

@@ -329,11 +329,11 @@ EOS
     lazy = [1, 2, 3].lazy
     assert_equal 3, lazy.size
     assert_equal 42, Enumerator.new(42){}.lazy.size
-    %i[map collect flat_map collect_concat].each do |m|
+    %i[map collect].each do |m|
       assert_equal 3, lazy.send(m){}.size
     end
     assert_equal 3, lazy.zip([4]).size
-    %i[select find_all reject take_while drop_while].each do |m|
+    %i[flat_map collect_concat select find_all reject take_while drop_while].each do |m|
       assert_equal nil, lazy.send(m){}.size
     end
     assert_equal nil, lazy.grep(//).size