Add while loop test to test_zjit.rb

1 files changed, 20 insertions, 0 deletions

diff --git a/test/ruby/test_zjit.rb b/test/ruby/test_zjit.rb
index 5265aac246..694a5032da 100644
--- a/test/ruby/test_zjit.rb
+++ b/test/ruby/test_zjit.rb
@@ -73,6 +73,26 @@ class TestZJIT < Test::Unit::TestCase
     }, call_threshold: 2
   end
 
+  def test_less_than
+    assert_compiles 'true', %q{
+      def test(a, b) = a < b
+      test(2, 5)
+    }, call_threshold: 2
+  end
+
+  def test_while_loop
+    assert_compiles '10', %q{
+      def loop_fun(n)
+        i = 0
+        while i < n
+          i = i + 1
+        end
+        i
+      end
+      loop_fun(10)
+    }, call_threshold: 2
+  end
+
   private
 
   # Assert that every method call in `test_script` can be compiled by ZJIT