Fix the previous commit for multiple arguments

[Fix GH-303]
SciRuby · Jun 22, 2021 · a19e3f3 · a19e3f3
1 parent d599c29
commit a19e3f3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/lib/iruby/backend.rb b/lib/iruby/backend.rb
@@ -65,8 +65,8 @@ def complete(code)
     def init_main_object(main)
       wrapper_module = Module.new
       main.extend(wrapper_module)
-      main.define_singleton_method(:include) do |mod|
-        wrapper_module.include(mod)
+      main.define_singleton_method(:include) do |*args, **kwargs|
+        wrapper_module.include(*args, **kwargs)
       end
     end
   end

diff --git a/test/iruby/backend_test.rb b/test/iruby/backend_test.rb
@@ -10,7 +10,7 @@ def test_eval_one_plus_two
 
     def test_include_module
       assert_nothing_raised do
-        @plainbackend.eval("include Math", false)
+        @plainbackend.eval("include Math, Comparable", false)
       end
     end
 
@@ -34,7 +34,7 @@ def test_eval_one_plus_two
 
     def test_include_module
       assert_nothing_raised do
-        @prybackend.eval("include Math", false)
+        @prybackend.eval("include Math, Comparable", false)
       end
     end