* enum.c (enum_take_while): returns Enumerator if no block given.

* enum.c (enum_drop_while): ditto.

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

1 files changed, 2 insertions, 0 deletions

diff --git a/enum.c b/enum.c
index b778bfea84..4196802de9 100644
--- a/enum.c
+++ b/enum.c
@@ -1451,6 +1451,7 @@ enum_take_while(VALUE obj)
 {
     VALUE ary;
 
+    RETURN_ENUMERATOR(obj, 0, 0);
     ary = rb_ary_new();
     rb_block_call(obj, id_each, 0, 0, take_while_i, (VALUE)&ary);
     return ary;
@@ -1522,6 +1523,7 @@ enum_drop_while(VALUE obj)
 {
     VALUE args[2];
 
+    RETURN_ENUMERATOR(obj, 0, 0);
     args[0] = rb_ary_new();
     args[1] = Qfalse;
     rb_block_call(obj, id_each, 0, 0, drop_while_i, (VALUE)args);