1 files changed, 97 insertions, 1 deletions
diff --git a/test/test_mathn.rb b/test/test_mathn.rb
index f511adb7d5..42a93139fd 100644
--- a/test/test_mathn.rb
+++ b/test/test_mathn.rb
@@ -5,6 +5,102 @@ require_relative 'ruby/envutil'
 class TestMathn < Test::Unit::TestCase
   def test_power
     assert_in_out_err ['-r', 'mathn', '-e', 'a=1**2;!a'], "", [], [], '[ruby-core:25740]'
-    assert_in_out_err ['-r', 'mathn', '-e', 'a=(1<<126)**2;!a'], "", [], [], '[ruby-core:25740]'
+    assert_in_out_err ['-r', 'mathn', '-e', 'a=(1 << 126)**2;!a'], "", [], [], '[ruby-core:25740]'
+  end
+
+  def assert_separated_equal(options, expected, actual, message = nil)
+    assert_in_out_err([*options, '-e', "p((#{actual})==(#{expected}))"], "", ["true"], [], message)
+  end
+
+  def test_floor
+    assert_separated_equal(%w[-rmathn], " 2", "( 13/5).floor")
+    assert_separated_equal(%w[-rmathn], " 2", "(  5/2).floor")
+    assert_separated_equal(%w[-rmathn], " 2", "( 12/5).floor")
+    assert_separated_equal(%w[-rmathn], "-3", "(-12/5).floor")
+    assert_separated_equal(%w[-rmathn], "-3", "( -5/2).floor")
+    assert_separated_equal(%w[-rmathn], "-3", "(-13/5).floor")
+
+    assert_separated_equal(%w[-rmathn], " 2", "( 13/5).floor(0)")
+    assert_separated_equal(%w[-rmathn], " 2", "(  5/2).floor(0)")
+    assert_separated_equal(%w[-rmathn], " 2", "( 12/5).floor(0)")
+    assert_separated_equal(%w[-rmathn], "-3", "(-12/5).floor(0)")
+    assert_separated_equal(%w[-rmathn], "-3", "( -5/2).floor(0)")
+    assert_separated_equal(%w[-rmathn], "-3", "(-13/5).floor(0)")
+
+    assert_separated_equal(%w[-rmathn], "( 13/5)", "( 13/5).floor(2)")
+    assert_separated_equal(%w[-rmathn], "(  5/2)", "(  5/2).floor(2)")
+    assert_separated_equal(%w[-rmathn], "( 12/5)", "( 12/5).floor(2)")
+    assert_separated_equal(%w[-rmathn], "(-12/5)", "(-12/5).floor(2)")
+    assert_separated_equal(%w[-rmathn], "( -5/2)", "( -5/2).floor(2)")
+    assert_separated_equal(%w[-rmathn], "(-13/5)", "(-13/5).floor(2)")
+  end
+
+  def test_ceil
+    assert_separated_equal(%w[-rmathn], " 3", "( 13/5).ceil")
+    assert_separated_equal(%w[-rmathn], " 3", "(  5/2).ceil")
+    assert_separated_equal(%w[-rmathn], " 3", "( 12/5).ceil")
+    assert_separated_equal(%w[-rmathn], "-2", "(-12/5).ceil")
+    assert_separated_equal(%w[-rmathn], "-2", "( -5/2).ceil")
+    assert_separated_equal(%w[-rmathn], "-2", "(-13/5).ceil")
+
+    assert_separated_equal(%w[-rmathn], " 3", "( 13/5).ceil(0)")
+    assert_separated_equal(%w[-rmathn], " 3", "(  5/2).ceil(0)")
+    assert_separated_equal(%w[-rmathn], " 3", "( 12/5).ceil(0)")
+    assert_separated_equal(%w[-rmathn], "-2", "(-12/5).ceil(0)")
+    assert_separated_equal(%w[-rmathn], "-2", "( -5/2).ceil(0)")
+    assert_separated_equal(%w[-rmathn], "-2", "(-13/5).ceil(0)")
+
+    assert_separated_equal(%w[-rmathn], "( 13/5)", "( 13/5).ceil(2)")
+    assert_separated_equal(%w[-rmathn], "(  5/2)", "(  5/2).ceil(2)")
+    assert_separated_equal(%w[-rmathn], "( 12/5)", "( 12/5).ceil(2)")
+    assert_separated_equal(%w[-rmathn], "(-12/5)", "(-12/5).ceil(2)")
+    assert_separated_equal(%w[-rmathn], "( -5/2)", "( -5/2).ceil(2)")
+    assert_separated_equal(%w[-rmathn], "(-13/5)", "(-13/5).ceil(2)")
+  end
+
+  def test_truncate
+    assert_separated_equal(%w[-rmathn], " 2", "( 13/5).truncate")
+    assert_separated_equal(%w[-rmathn], " 2", "(  5/2).truncate")
+    assert_separated_equal(%w[-rmathn], " 2", "( 12/5).truncate")
+    assert_separated_equal(%w[-rmathn], "-2", "(-12/5).truncate")
+    assert_separated_equal(%w[-rmathn], "-2", "( -5/2).truncate")
+    assert_separated_equal(%w[-rmathn], "-2", "(-13/5).truncate")
+
+    assert_separated_equal(%w[-rmathn], " 2", "( 13/5).truncate(0)")
+    assert_separated_equal(%w[-rmathn], " 2", "(  5/2).truncate(0)")
+    assert_separated_equal(%w[-rmathn], " 2", "( 12/5).truncate(0)")
+    assert_separated_equal(%w[-rmathn], "-2", "(-12/5).truncate(0)")
+    assert_separated_equal(%w[-rmathn], "-2", "( -5/2).truncate(0)")
+    assert_separated_equal(%w[-rmathn], "-2", "(-13/5).truncate(0)")
+
+    assert_separated_equal(%w[-rmathn], "( 13/5)", "( 13/5).truncate(2)")
+    assert_separated_equal(%w[-rmathn], "(  5/2)", "(  5/2).truncate(2)")
+    assert_separated_equal(%w[-rmathn], "( 12/5)", "( 12/5).truncate(2)")
+    assert_separated_equal(%w[-rmathn], "(-12/5)", "(-12/5).truncate(2)")
+    assert_separated_equal(%w[-rmathn], "( -5/2)", "( -5/2).truncate(2)")
+    assert_separated_equal(%w[-rmathn], "(-13/5)", "(-13/5).truncate(2)")
+  end
+
+  def test_round
+    assert_separated_equal(%w[-rmathn], " 3", "( 13/5).round")
+    assert_separated_equal(%w[-rmathn], " 3", "(  5/2).round")
+    assert_separated_equal(%w[-rmathn], " 2", "( 12/5).round")
+    assert_separated_equal(%w[-rmathn], "-2", "(-12/5).round")
+    assert_separated_equal(%w[-rmathn], "-3", "( -5/2).round")
+    assert_separated_equal(%w[-rmathn], "-3", "(-13/5).round")
+
+    assert_separated_equal(%w[-rmathn], " 3", "( 13/5).round(0)")
+    assert_separated_equal(%w[-rmathn], " 3", "(  5/2).round(0)")
+    assert_separated_equal(%w[-rmathn], " 2", "( 12/5).round(0)")
+    assert_separated_equal(%w[-rmathn], "-2", "(-12/5).round(0)")
+    assert_separated_equal(%w[-rmathn], "-3", "( -5/2).round(0)")
+    assert_separated_equal(%w[-rmathn], "-3", "(-13/5).round(0)")
+
+    assert_separated_equal(%w[-rmathn], "( 13/5)", "( 13/5).round(2)")
+    assert_separated_equal(%w[-rmathn], "(  5/2)", "(  5/2).round(2)")
+    assert_separated_equal(%w[-rmathn], "( 12/5)", "( 12/5).round(2)")
+    assert_separated_equal(%w[-rmathn], "(-12/5)", "(-12/5).round(2)")
+    assert_separated_equal(%w[-rmathn], "( -5/2)", "( -5/2).round(2)")
+    assert_separated_equal(%w[-rmathn], "(-13/5)", "(-13/5).round(2)")
   end
 end