diff options
| -rw-r--r-- | mrbgems/mruby-enumerator/mrblib/enumerator.rb | 10 | ||||
| -rw-r--r-- | mrbgems/mruby-enumerator/test/enumerator.rb | 4 |
2 files changed, 13 insertions, 1 deletions
diff --git a/mrbgems/mruby-enumerator/mrblib/enumerator.rb b/mrbgems/mruby-enumerator/mrblib/enumerator.rb index e1c7d7ef5..7915a8472 100644 --- a/mrbgems/mruby-enumerator/mrblib/enumerator.rb +++ b/mrbgems/mruby-enumerator/mrblib/enumerator.rb @@ -216,7 +216,15 @@ class Enumerator def inspect return "#<#{self.class}: uninitialized>" unless @obj - "#<#{self.class}: #{@obj}:#{@meth}>" + + args = "" + if @args && @args.size > 0 + args = "(" + @args.each {|arg| args << "#{arg}, " } + args = args[0, args.size-2] + args << ")" + end + "#<#{self.class}: #{@obj}:#{@meth}#{args}>" end ## diff --git a/mrbgems/mruby-enumerator/test/enumerator.rb b/mrbgems/mruby-enumerator/test/enumerator.rb index 5954af4e0..b22ed7754 100644 --- a/mrbgems/mruby-enumerator/test/enumerator.rb +++ b/mrbgems/mruby-enumerator/test/enumerator.rb @@ -90,6 +90,10 @@ end assert 'Enumerator#inspect' do e = (0..10).each assert_equal("#<Enumerator: 0..10:each>", e.inspect) + e = Enumerator.new("FooObject", :foo, 1) + assert_equal("#<Enumerator: FooObject:foo(1)>", e.inspect) + e = Enumerator.new("FooObject", :foo, 1, 2, 3) + assert_equal("#<Enumerator: FooObject:foo(1, 2, 3)>", e.inspect) end assert 'Enumerator#each' do |