Added median width to box plot and improved tests

Author: alishdipani

lib/rubyplot/artist/plot/box_plot.rb

@@ -14,15 +14,17 @@ class BoxPlot < Artist::Plot::Base
         attr_accessor :outlier_marker_type
         attr_accessor :outlier_marker_color
         attr_accessor :outlier_marker_size
-        
+        attr_accessor :median_width
+
         def initialize(*)
           super
           @whiskers = 1.5
           @x_left_box = []
-          @median_color = :yellow
+          @median_color = :black
           @outlier_marker_type = :plus
           @outlier_marker_color = nil
           @outlier_marker_size = 1.0
+          @median_width = 3.0
         end
 
         def process_data
@@ -45,7 +47,7 @@ def draw
             draw_box x_left, i
             draw_whiskers x_left, i
             draw_outliers x_left, i
-            draw_median x_left, i    
+            draw_median x_left, i
           end
         end
 
@@ -111,7 +113,7 @@ def draw_outliers x_left, index
 
         def first_max_outlier_index vector, max
           return nil if vector.last >= max
-          vector.size.times do |i| 
+          vector.size.times do |i|
             if vector[i] > max
               return i
             end
@@ -131,7 +133,8 @@ def draw_median x_left, index
           Rubyplot::Artist::Line2D.new(self,
             x: [x_left, x_left + @box_width],
             y: [@medians[index], @medians[index]],
-            color: @median_color
+            color: @median_color,
+            width: @median_width
           ).draw
         end
 
@@ -141,7 +144,7 @@ def calculate_ranges!
           @medians = []
           @mins = []
           @maxs = []
-          
+
           @vectors.each do |sorted_vec|
             m = get_percentile 50, sorted_vec
             q1 = get_percentile 25, sorted_vec

spec/axes_spec.rb

@@ -901,6 +901,26 @@
       axes.y_title = "bar"
     end
 
+    it "adds a simple box plot with outliers" do
+      @figure = Rubyplot::Figure.new
+      axes = @figure.add_subplot! 0,0
+      axes.title = "A simple box plot."
+      axes.box_plot! do |p|
+        p.data [
+          [60,70,80,70,50],
+          [100,40,20,80,70],
+          [30, 10]
+        ]
+        p.whiskers = 0.1
+        p.outlier_marker_type = :diamond
+        p.outlier_marker_size = 3.0
+        p.outlier_marker_color = :red
+        p.median_width = 4.0
+      end
+      axes.x_title = "foo"
+      axes.y_title = "bar"
+    end
+
     it "adds a simple horizontal box plot" do
       skip "Leave for after initial box plot setup is complete."
     end
@@ -913,7 +933,7 @@
         p.data [
           [-48, 2,60,70,80,70,50],
           [4,100,40,20,80,70],
-          [1,30, 10]          
+          [1,30, 10]
         ]
       end
       axes.box_plot! do |p|
@@ -935,7 +955,7 @@
         p.data [
           [-48,60,70,80,70,50],
           [4,100,40,20,80,70],
-          [1,30, 10]          
+          [1,30, 10]
         ]
         p.whiskers = 0.3
       end
@@ -954,25 +974,25 @@
 
   context "#top_margin=" do
     it "sets the top margin in pixels" do
-      
+
     end
   end
 
   context "#left_margin=" do
     it "sets the left margin in pixels" do
-      
+
     end
   end
 
   context "#bottom_margin=" do
     it "sets the bottom margin in pixels" do
-      
+
     end
   end
 
   context "#right_margin=" do
     it "sets the right margin in pixels" do
-      
+
     end
   end
 
@@ -988,4 +1008,3 @@
     end
   end # context "#x_ticks="
 end # Rubyplot::Axes
-