test_gnuplot.rb

# -*- ruby -*-

require 'gnuplot'
require 'minitest/autorun'

class StdDataTest < MiniTest::Test

  def test_array_1d
    data = (0..5).to_a
    ds = Gnuplot::DataSet.new( data )

    assert data == ds.data
    assert data.join("\n") + "\n", ds.to_gplot
  end

  def test_empty_array
    data = []
    ds = Gnuplot::DataSet.new( data )

    assert data == ds.data
    assert "" == ds.to_gplot
  end


  # Test a multidimensional array.

  def test_array_nd
    d1 = (0..3).to_a
    d2 = d1.collect { |v| 3 * v }
    d3 = d2.collect { |v| 4 * v }

    data = [ d1, d2, d3 ]
    ds = Gnuplot::DataSet.new( data )

    assert data == ds.data
    assert "0 0 0\n1 3 12\n2 6 24\n3 9 36\n", ds.to_gplot
  end
end


class DataSetTest < MiniTest::Test

  def test_yield_ctor
    ds = Gnuplot::DataSet.new do |ds|
      ds.with = "lines"
      ds.using = "1:2"
      ds.data = [ [0, 1, 2], [1, 2, 5] ]
    end

    assert "lines", ds.with
    assert "1:2",   ds.using
    assert nil == ds.title
    assert [ [0, 1, 2], [1, 2, 5] ] == ds.data
    assert "'-' using 1:2 with lines",  ds.plot_args
    assert "0 1\n1 2\n2 5\n", ds.to_gplot
  end

end


class PlotTest < MiniTest::Test

  def test_no_data
    plot = Gnuplot::Plot.new do |p|
      p.set "output", "'foo'"
      p.set "terminal", "postscript enhanced"
      p.unset "border"
    end

    assert( plot.settings ==
		 [ [:set, "output", "'foo'"],
		   [:set, "terminal", "postscript enhanced"],
                   [:unset, "border"] ] )


    assert( plot.to_gplot, \
		 "set output 'foo'\nset terminal postscript enhanced\n" )

  end

  def test_set
    plot = Gnuplot::Plot.new do |p|
      p.set "title", "foo"
    end
    assert "'foo'", plot["title"]

    plot.set "title", "'foo'"
    assert "'foo'", plot["title"]
  end

  def test_unset
    plot = Gnuplot::Plot.new do |p|
      p.unset "title"
    end
    assert_nil plot["title"]

    plot.unset "title"
    assert_nil plot["title"]
  end

  def test_style
    plot = Gnuplot::Plot.new do |p|
      s1 = p.style do |s|
        s.ls = 1
        s.lt = 1
        s.lc = 1
        s.pt = 1
        s.ps = 1
      end
      assert s1.to_s == "set style line 1 ls 1 lt 1 lc 1 pt 1 ps 1", "correct style definition"
      assert s1.index == 1, "set index correctly"

      s2 = p.style do |s|
        s.ls = 2
      end
      assert s2.to_s == "set style line 2 ls 2", "correct style definition"
      assert s2.index == 2, "index must be incremented"

      ds = Gnuplot::DataSet.new do |ds|
        ds.with = "lines"
        ds.linestyle = s1
        ds.data = [ [0, 1, 2], [1, 2, 5] ]
      end

      assert ds.plot_args == "'-' with lines linestyle 1", "convert linestyle to index" # index of s1
    end
  end

end


require 'rbconfig'
CONFIG = RbConfig::MAKEFILE_CONFIG

# This attempts to test the functions that comprise the gnuplot package.  Most
# of the bug reports that I get for this package have to do with finding the
# gnuplot executable under different environments so that makes it difficult
# to test on a single environment.  To try to get around this I'm using the
# rbconfig library and its path to the sh environment variable.

class GnuplotModuleTest

  def test_which
    # Put the spaces around the command to make sure that it gets stripped
    # properly.
    assert( CONFIG["SHELL"], Gnuplot::which(" sh " ) )
    assert( CONFIG["SHELL"], Gnuplot::which( CONFIG["SHELL"] ) )
  end


  def test_gnuplot
    cmd = Gnuplot.gnuplot
    assert( Gnuplot::which("gnuplot") + " -persist", cmd )

    cmd = Gnuplot.gnuplot(false)
    assert( Gnuplot::which("gnuplot"), cmd )

    # If I set the name of the gnuplot environment variable to a different
    # name (one that is in the path) then I should get the shell name as the
    # result of the gnuplot call.

    ENV["RB_GNUPLOT"] = "sh"
    assert( CONFIG["SHELL"], Gnuplot.gnuplot(false) )
  end

end