gruff 0.4.0-java

data/test/test_scatter.rb

@@ -0,0 +1,233 @@
+#!/usr/bin/ruby
+
+require File.dirname(__FILE__) + '/gruff_test_case'
+
+class TestGruffScatter < Test::Unit::TestCase
+
+  def setup
+    @datasets = [
+        [:Chuck, [20, 10, 5, 12, 11, 6, 10, 7], [5, 10, 19, 6, 9, 1, 14, 8]],
+        [:Brown, [5, 10, 20, 6, 9, 12, 14, 8], [20, 10, 5, 12, 11, 6, 10, 7]],
+        [:Lucy, [19, 9, 6, 11, 12, 7, 15, 8], [6, 11, 18, 8, 12, 8, 10, 6]]
+    ]
+  end
+
+  # Done
+  def test_scatter_graph
+    g = setup_basic_graph
+    g.title = 'Basic Scatter Plot Test'
+    g.write('test/output/scatter_basic.png')
+  end
+
+  #~ # Done
+  def test_many_datapoints
+    g = Gruff::Scatter.new
+    g.title = 'Many Datapoint Graph Test'
+    y_values = (0..50).map { rand(100) }
+    x_values = (0..50).map { rand(100) }
+    g.data('many points', y_values, x_values)
+
+    # Default theme
+    g.write('test/output/scatter_many.png')
+  end
+
+  # Done
+  def test_no_data
+    g = Gruff::Scatter.new(400)
+    g.title = 'No Data'
+    # Default theme
+    g.write('test/output/scatter_no_data.png')
+
+    g = Gruff::Scatter.new(400)
+    g.title = 'No Data Title'
+    g.no_data_message = 'There is no data'
+    g.write('test/output/scatter_no_data_msg.png')
+  end
+
+  # Done
+  def test_all_zeros
+    g = Gruff::Scatter.new(400)
+    g.title = 'All Zeros'
+
+    g.data(:gus, [0, 0, 0, 0], [0, 0, 0, 0])
+
+    # Default theme
+    g.write('test/output/scatter_no_data_other.png')
+  end
+
+  # Done
+  def test_some_nil_points
+    g = Gruff::Scatter.new
+    g.title = 'Some Nil Points'
+
+    @datasets = [
+        [:data1, [1, 2, 3, nil, 3, 5, 6], [5, nil, nil, nil, nil, 5, 7]]
+    ]
+
+    @datasets.each do |data|
+      assert_raise ArgumentError do
+        g.data(*data)
+      end
+    end
+  end
+
+  # Done
+  def test_unequal_number_of_x_and_y_values
+    g = Gruff::Scatter.new
+    g.title = 'Unequal number of X and Y values'
+
+    @datasets = [
+        [:data1, [1, 2, 3], [1, 2]],
+        [:data2, [1, 2, 3, 4, 5], [1, 2, 3, 4, 5, 6]],
+    ]
+
+    @datasets.each do |data|
+      assert_raise ArgumentError do
+        g.data(*data)
+      end
+    end
+  end
+
+  # Done
+  def test_empty_set_of_axis_values
+    g = Gruff::Scatter.new
+    g.title = 'Missing Axis Values'
+
+    @datasets = [
+        [:data1, [1, 2, 3, 4, 5]]
+    ]
+
+    @datasets.each do |data|
+      assert_raise ArgumentError do
+        g.data(*data)
+      end
+    end
+  end
+
+  # Done
+  def test_no_title
+    g = Gruff::Scatter.new(400)
+    g.data(:data1, [1, 2, 3, 4, 5], [1, 2, 3, 4, 5])
+    g.write('test/output/scatter_no_title.png')
+  end
+
+  # Done
+  def test_no_line_markers
+    g = setup_basic_graph(400)
+    g.title = 'No Line Markers'
+    g.hide_line_markers = true
+    g.write('test/output/scatter_no_line_markers.png')
+  end
+
+  # Done
+  def test_no_legend
+    g = setup_basic_graph(400)
+    g.title = 'No Legend'
+    g.hide_legend = true
+    g.write('test/output/scatter_no_legend.png')
+  end
+
+  # Done
+  def test_nothing_but_the_graph
+    g = setup_basic_graph(400)
+    g.title = 'THIS TITLE SHOULD NOT DISPLAY!!!'
+    g.hide_line_markers = true
+    g.hide_legend = true
+    g.hide_title = true
+    g.write('test/output/scatter_nothing_but_the_graph.png')
+  end
+
+  # TODO Implement baselines on x and y axis
+  #~ def test_baseline_larger_than_data
+  #~ g = setup_basic_graph(400)
+  #~ g.title = "Baseline Larger Than Data"
+  #~ g.baseline_value = 150
+  #~ g.write("test/output/scatter_large_baseline.png")
+  #~ end
+
+  # Done
+  def test_wide_graph
+    g = setup_basic_graph('800x400')
+    g.title = 'Wide Graph'
+    g.write('test/output/scatter_wide_graph.png')
+
+    g = setup_basic_graph('400x200')
+    g.title = 'Wide Graph Small'
+    g.write('test/output/scatter_wide_graph_small.png')
+  end
+
+  # Done
+  def test_negative
+    g = setup_pos_neg(800)
+    g.write('test/output/scatter_pos_neg.png')
+
+    g = setup_pos_neg(400)
+    g.title = 'Pos/Neg Line Test Small'
+    g.write('test/output/scatter_pos_neg_400.png')
+  end
+
+  # Done
+  def test_all_negative
+    g = setup_all_neg(800)
+    g.write('test/output/scatter_all_neg.png')
+
+    g = setup_all_neg(400)
+    g.title = 'All Neg Line Test Small'
+    g.write('test/output/scatter_all_neg_400.png')
+  end
+
+  # Done
+  def test_no_hide_line_no_labels
+    g = Gruff::Scatter.new
+    g.title = 'No Hide Line No Labels'
+    @datasets.each do |data|
+      g.data(data[0], data[1], data[2])
+    end
+    g.hide_line_markers = false
+    g.write('test/output/scatter_no_hide.png')
+  end
+
+  def test_no_set_labels
+    g = Gruff::Scatter.new
+    g.title = 'Setting Labels Test'
+    g.labels = {
+        0 => 'This',
+        1 => 'should',
+        2 => 'not',
+        3 => 'show',
+        4 => 'up'
+    }
+    @datasets.each do |data|
+      g.data(data[0], data[1], data[2])
+    end
+    g.write('test/output/scatter_no_labels.png')
+  end
+
+  protected
+
+  def setup_basic_graph(size=800)
+    g = Gruff::Scatter.new(size)
+    g.title = 'Rad Graph'
+    @datasets.each do |data|
+      g.data(data[0], data[1], data[2])
+    end
+    g
+  end
+
+  def setup_pos_neg(size=800)
+    g = Gruff::Scatter.new(size)
+    g.title = 'Pos/Neg Scatter Graph Test'
+    g.data(:apples, [-1, 0, 4, -4], [-5, -1, 3, 4])
+    g.data(:peaches, [10, 8, 6, 3], [-1, 1, 3, 3])
+    g
+  end
+
+  def setup_all_neg(size=800)
+    g = Gruff::Scatter.new(size)
+    g.title = 'Neg Scatter Graph Test'
+    g.data(:apples, [-1, -1, -4, -4], [-5, -1, -3, -4])
+    g.data(:peaches, [-10, -8, -6, -3], [-1, -1, -3, -3])
+    g
+  end
+
+end # end GruffTestCase

data/test/test_scene.rb

@@ -0,0 +1,100 @@
+#!/usr/bin/ruby
+
+require File.dirname(__FILE__) + "/gruff_test_case"
+require 'yaml'
+
+class LayerStub < Gruff::Layer; attr_reader :base_dir, :filenames, :selected_filename; end
+
+class TestGruffScene < GruffTestCase
+
+  def test_hazy
+    g = setup_scene
+    g.weather = "cloudy"
+    g.haze = true
+    g.time = Time.mktime(2006, 7, 4, 4, 35)
+    g.write "test/output/scene_hazy_night.png"
+  end
+
+  def test_stormy_night
+    g = setup_scene
+    g.weather = "stormy"
+    g.time = Time.mktime(2006, 7, 4, 0, 0)
+    g.write "test/output/scene_stormy_night.png"
+  end
+
+
+  def test_not_hazy
+    g = setup_scene
+    g.weather = "cloudy"
+    g.haze = false
+    g.time = Time.mktime(2006, 7, 4, 6, 00)
+    g.write "test/output/scene_not_hazy_day.png"
+  end
+
+  def test_partly_cloudy
+    g = setup_scene
+    g.weather = "partly cloudy"
+    g.haze = false
+    g.time = Time.mktime(2006, 7, 4, 13, 00)
+    g.write "test/output/scene_partly_cloudy_day.png"
+  end
+
+
+  def test_stormy_day
+    g = setup_scene
+    g.weather = "stormy"
+    g.haze = false
+    g.time = Time.mktime(2006, 7, 4, 8, 00)
+    g.write "test/output/scene_stormy_day.png"
+  end
+
+
+  def test_layer
+    l = LayerStub.new(File.expand_path("../assets/city_scene", File.dirname(__FILE__)), "clouds")
+    assert_equal %w(cloudy.png partly_cloudy.png stormy.png), l.filenames.sort
+    
+    l = LayerStub.new(File.expand_path("../assets/city_scene", File.dirname(__FILE__)), "grass")
+    assert_equal 'default.png', l.selected_filename
+    
+    l = LayerStub.new(File.expand_path("../assets/city_scene", File.dirname(__FILE__)), "sky")
+    l.update Time.mktime(2006, 7, 4, 12, 35) # 12:35, July 4, 2006
+    assert_equal '1200.png', l.selected_filename
+
+    l = LayerStub.new(File.expand_path("../assets/city_scene", File.dirname(__FILE__)), "sky")
+    l.update Time.mktime(2006, 7, 4, 0, 0) # 00:00, July 4, 2006
+    assert_equal '0000.png', l.selected_filename
+
+    l = LayerStub.new(File.expand_path("../assets/city_scene", File.dirname(__FILE__)), "sky")
+    l.update Time.mktime(2006, 7, 4, 23, 35) # 23:35, July 4, 2006
+    assert_equal '2000.png', l.selected_filename
+
+    l = LayerStub.new(File.expand_path("../assets/city_scene", File.dirname(__FILE__)), "sky")
+    l.update Time.mktime(2006, 7, 4, 0, 1) # 00:01, July 4, 2006
+    assert_equal '0000.png', l.selected_filename
+
+    l = LayerStub.new(File.expand_path("../assets/city_scene", File.dirname(__FILE__)), "sky")
+    l.update Time.mktime(2006, 7, 4, 2, 0) # 02:00, July 4, 2006
+    assert_equal '0200.png', l.selected_filename
+
+    l = LayerStub.new(File.expand_path("../assets/city_scene", File.dirname(__FILE__)), "sky")
+    l.update Time.mktime(2006, 7, 4, 4, 00) # 04:00, July 4, 2006
+    assert_equal '0400.png', l.selected_filename
+    
+    # TODO Need number_sample folder
+    # l = LayerStub.new(File.expand_path("../assets/city_scene", File.dirname(__FILE__)), "number_sample")
+    # assert_equal %w(1.png 2.png default.png), l.filenames
+    # l.update 3
+    # assert_equal 'default.png', l.selected_filename
+  end
+
+private
+
+  def setup_scene
+    g = Gruff::Scene.new("500x100", File.expand_path("../assets/city_scene", File.dirname(__FILE__)) )
+    g.layers = %w(background haze sky clouds)
+    g.weather_group = %w(clouds)
+    g.time_group = %w(background sky)
+    g    
+  end
+
+end

data/test/test_side_bar.rb

@@ -0,0 +1,56 @@
+
+require File.dirname(__FILE__) + "/gruff_test_case"
+
+class TestGruffSideBar < GruffTestCase
+
+  def test_bar_graph
+    g = setup_basic_graph(Gruff::SideBar, 800)
+    write_test_file g, 'side_bar.png'    
+  end
+
+  def test_bar_spacing
+    g = setup_basic_graph(Gruff::SideBar, 800)
+    g.bar_spacing = 0
+    g.title = "100% spacing between bars"
+    g.write("test/output/side_bar_spacing_full.png")
+
+    g = setup_basic_graph(Gruff::SideBar, 800)
+    g.bar_spacing = 0.5
+    g.title = "50% spacing between bars"
+    g.write("test/output/side_bar_spacing_half.png")
+
+    g = setup_basic_graph(Gruff::SideBar, 800)
+    g.bar_spacing = 1
+    g.title = "0% spacing between bars"
+    g.write("test/output/side_bar_spacing_none.png")
+  end
+  
+  def test_x_axis_range
+    g = Gruff::SideBar.new('400x300')
+    g.title = 'Should run from 8 to 32'
+    g.hide_line_numbers = false
+    g.theme_37signals
+    g.data("Grapes", [8])
+    g.data("Apples", [24])
+    g.data("Oranges", [32])
+    g.data("Watermelon", [8])
+    g.data("Peaches", [12])
+    g.labels = {0 => '2003', 2 => '2004', 4 => '2005'}
+    g.write("test/output/side_bar_data_range.png")
+  end
+  
+  def test_bar_labels
+    g = Gruff::SideBar.new('400x300')
+    g.title = 'Should show labels for each bar'
+    g.data("Grapes", [8])
+    g.data("Apples", [24])
+    g.data("Oranges", [32])
+    g.data("Watermelon", [8])
+    g.data("Peaches", [12])
+    g.labels = {0 => '2003', 2 => '2004', 4 => '2005'}
+    g.show_labels_for_bar_values = true
+    g.write("test/output/side_bar_labels.png")
+  end
+
+end
+

data/test/test_sidestacked_bar.rb

@@ -0,0 +1,105 @@
+#!/usr/bin/ruby
+
+require File.dirname(__FILE__) + "/gruff_test_case"
+
+class TestGruffSideStackedBar < GruffTestCase
+
+  def setup
+    @datasets = [
+      [:Jimmy, [25, 36, 86, 39]],
+      [:Charles, [80, 54, 67, 54]],
+      [:Julie, [22, 29, 35, 38]],
+      #[:Jane, [95, 95, 95, 90, 85, 80, 88, 100]],
+      #[:Philip, [90, 34, 23, 12, 78, 89, 98, 88]],
+      #["Arthur", [5, 10, 13, 11, 6, 16, 22, 32]],
+      ]
+    @sample_labels = {
+        0 => '5/6', 
+        1 => '5/15', 
+        2 => '5/24'
+      }      
+
+  end
+
+  def test_bar_graph
+    g = Gruff::SideStackedBar.new
+    g.title = "Visual Stacked Bar Graph Test"
+    g.labels = {
+      0 => '5/6', 
+      1 => '5/15', 
+      2 => '5/24', 
+      3 => '5/30', 
+    }
+    @datasets.each do |data|
+      g.data(data[0], data[1])
+    end
+    g.write "test/output/side_stacked_bar_keynote.png"
+  end
+
+
+  def test_bar_graph_small
+    g = Gruff::SideStackedBar.new(400)
+    g.title = "Visual Stacked Bar Graph Test"
+    g.labels = {
+      0 => '5/6', 
+      1 => '5/15', 
+      2 => '5/24', 
+      3 => '5/30', 
+    }
+    @datasets.each do |data|
+      g.data(data[0], data[1])
+    end
+    g.write "test/output/side_stacked_bar_keynote_small.png"
+  end
+
+  def test_wide
+    g = setup_basic_graph('800x400')
+    g.title = "Wide SSBar"
+    g.write "test/output/side_stacked_bar_wide.png"
+  end
+
+  def test_should_space_long_left_labels_appropriately
+    g = Gruff::SideStackedBar.new
+    g.title = "Stacked Bar Long Label"
+    g.labels = {
+      0 => 'September', 
+      1 => 'Oct', 
+      2 => 'Nov', 
+      3 => 'Dec', 
+    }
+    @datasets.each do |data|
+      g.data(data[0], data[1])
+    end
+    g.write "test/output/side_stacked_bar_long_label.png"
+  end
+  
+  def test_bar_labels
+     g = Gruff::SideStackedBar.new
+     g.title = "Stacked Bar Long Label"
+     g.labels = {
+       0 => 'September', 
+       1 => 'Oct', 
+       2 => 'Nov', 
+       3 => 'Dec', 
+     }
+     @datasets.each do |data|
+       g.data(data[0], data[1])
+     end
+     g.show_labels_for_bar_values = true
+     g.write "test/output/side_stacked_bar_labels.png"
+   end
+
+protected
+
+  def setup_basic_graph(size=800)
+    g = Gruff::SideStackedBar.new(size)
+    g.title = "My Graph Title"
+    g.labels = @sample_labels
+    @datasets.each do |data|
+      g.data(data[0], data[1])
+    end
+    return g
+  end
+
+end
+