diff --git a/activesupport/lib/active_support/core_ext/float.rb b/activesupport/lib/active_support/core_ext/float.rb new file mode 100644 index 0000000000000000000000000000000000000000..ac5d5a47e63f4d9d0175b7261bdc2ca3c67e3f13 --- /dev/null +++ b/activesupport/lib/active_support/core_ext/float.rb @@ -0,0 +1,5 @@ +require File.dirname(__FILE__) + '/float/rounding' + +class Float #:nodoc: + include ActiveSupport::CoreExtensions::Float::Rounding +end diff --git a/activesupport/lib/active_support/core_ext/float/rounding.rb b/activesupport/lib/active_support/core_ext/float/rounding.rb new file mode 100644 index 0000000000000000000000000000000000000000..989b1f8f809f13c4a169482b034e276a2edb2a8b --- /dev/null +++ b/activesupport/lib/active_support/core_ext/float/rounding.rb @@ -0,0 +1,22 @@ +module ActiveSupport #:nodoc: + module CoreExtensions #:nodoc: + module Float #:nodoc: + module Rounding + def self.included(base) #:nodoc: + base.send(:alias_method, :round_without_precision, :round) + base.send(:alias_method, :round, :round_with_precision) + end + + # Rounds the float with the specified precision. + # + # x = 1.337 + # x.round # => 1 + # x.round(1) # => 1.3 + # x.round(2) # => 1.34 + def round_with_precision(precision = nil) + precision.nil? ? round_without_precision : (self * (10 ** precision)).round / (10 ** precision).to_f + end + end + end + end +end diff --git a/activesupport/test/core_ext/float_ext_test.rb b/activesupport/test/core_ext/float_ext_test.rb new file mode 100644 index 0000000000000000000000000000000000000000..b74add519fbcdb6bae365aa90e3ee69d78c2036e --- /dev/null +++ b/activesupport/test/core_ext/float_ext_test.rb @@ -0,0 +1,25 @@ +require File.dirname(__FILE__) + '/../abstract_unit' + +class FloatExtRoundingTests < Test::Unit::TestCase + def test_round_for_positive_number + assert_equal 1, 1.4.round + assert_equal 2, 1.6.round + assert_equal 2, 1.6.round(0) + assert_equal 1.4, 1.4.round(1) + assert_equal 1.4, 1.4.round(3) + assert_equal 1.5, 1.45.round(1) + assert_equal 1.45, 1.445.round(2) + end + + def test_round_for_negative_number + assert_equal( -1, -1.4.round ) + assert_equal( -2, -1.6.round ) + assert_equal( -1.4, -1.4.round(1) ) + assert_equal( -1.5, -1.45.round(1) ) + end + + def test_round_with_negative_precision + assert_equal 123460.0, 123456.0.round(-1) + assert_equal 123500.0, 123456.0.round(-2) + end +end