$:.unshift(File.dirname(__FILE__) + "/../lib/")

require 'test/unit'
require 'gruff'

class GruffTestCase < Test::Unit::TestCase

  def test_dummy
    assert true
  end

protected

  # Generate graphs at several sizes.
  #
  # Also writes the graph to disk.
  #
  #   graph_sized 'bar_basic' do |g|
  #     g.data('students', [1, 2, 3, 4])
  #   end
  #
  def graph_sized(filename, &block)
    class_name = self.class.name.gsub(/^TestGruff/, '')
    ['', 400].each do |size|
      g = instance_eval("Gruff::#{class_name}.new #{size}")
      block.call g
      write_test_file g, "#{filename}#{size}.png"
    end
  end

  def write_test_file(graph, filename)
    graph.write(File.dirname(__FILE__) + "/output/#{filename}")
  end

end