Fix remove_instance_variable for too complex class

1 files changed, 18 insertions, 1 deletions

diff --git a/test/ruby/test_shapes.rb b/test/ruby/test_shapes.rb
index a098edf409..529964729b 100644
--- a/test/ruby/test_shapes.rb
+++ b/test/ruby/test_shapes.rb
@@ -233,7 +233,7 @@ class TestShapes < Test::Unit::TestCase
     end;
   end
 
-  def test_run_out_of_shape
+  def test_run_out_of_shape_for_object
     assert_separately([], "#{<<~"begin;"}\n#{<<~'end;'}")
     begin;
       class A
@@ -252,6 +252,23 @@ class TestShapes < Test::Unit::TestCase
     end;
   end
 
+  def test_run_out_of_shape_for_class
+    assert_separately([], "#{<<~"begin;"}\n#{<<~'end;'}")
+    begin;
+      c = Class.new
+      i = 0
+      while RubyVM::Shape.shapes_available > 0
+        c.instance_variable_set(:"@i#{i}", 1)
+        i += 1
+      end
+
+      c.instance_variable_set(:@a, 1)
+      assert_equal(1, c.instance_variable_get(:@a))
+      c.remove_instance_variable(:@a)
+      assert_nil(c.instance_variable_get(:@a))
+    end;
+  end
+
   def test_run_out_of_shape_generic_ivar_set
     assert_separately([], "#{<<~"begin;"}\n#{<<~'end;'}")
     begin;