lib/guard/rspec/formatter.rb in guard-rspec-0.1.9 vs lib/guard/rspec/formatter.rb in guard-rspec-0.2.0

@@ -1,29 +1,40 @@
 require "#{File.dirname(__FILE__)}/../rspec"
 
 module Formatter
-  
+
   def guard_message(example_count, failure_count, pending_count, duration)
     message = "#{example_count} examples, #{failure_count} failures"
     if pending_count > 0
       message << " (#{pending_count} pending)"
     end
-    message << "\nin #{duration} seconds"
+    message << "\nin #{round_float(duration)} seconds"
     message
   end
-  
+
   # failed | pending | success
   def guard_image(failure_count, pending_count)
     icon = if failure_count > 0
       :failed
     elsif pending_count > 0
       :pending
     else
       :success
     end
   end
-  
+
   def notify(message, image)
     Guard::Notifier.notify(message, :title => "RSpec results", :image => image)
   end
-  
-end
\ No newline at end of file
+
+private
+
+  def round_float(float, decimals=4)
+    if Float.instance_method(:round).arity == 0 # Ruby 1.8
+      factor = 10**decimals
+      (float*factor).round / factor.to_f
+    else # Ruby 1.9
+      float.round(decimals)
+    end
+  end
+
+end