Switch shape test to use exhaust_shapes

1 files changed, 4 insertions, 9 deletions

diff --git a/test/ruby/test_shapes.rb b/test/ruby/test_shapes.rb
index 7919a210a7..c4b0d251f1 100644
--- a/test/ruby/test_shapes.rb
+++ b/test/ruby/test_shapes.rb
@@ -132,17 +132,12 @@ class TestShapes < Test::Unit::TestCase
     begin;
       class Hi; end
 
-      obj = Hi.new
-      i = 0
-      while RubyVM::Shape.shapes_available > 2
-        obj.instance_variable_set(:"@a#{i}", 1)
-        i += 1
-      end
+      RubyVM::Shape.exhaust_shapes(2)
 
       obj = Hi.new
-      obj.instance_variable_set(:"@b", 1)
-      obj.instance_variable_set(:"@c", 1)
-      obj.instance_variable_set(:"@d", 1)
+      obj.instance_variable_set(:@b, 1)
+      obj.instance_variable_set(:@c, 1)
+      obj.instance_variable_set(:@d, 1)
 
       assert_predicate RubyVM::Shape.of(obj), :too_complex?
     end;