unicode_plot 0.0.2 → 0.0.3

data/test/test-canvas.rb

@@ -4,7 +4,9 @@ module CanvasTestCases
 
   CANVAS_CLASSES = {
     ascii: UnicodePlot::AsciiCanvas,
-    braille: UnicodePlot::BrailleCanvas
+    braille: UnicodePlot::BrailleCanvas,
+    density: UnicodePlot::DensityCanvas,
+    dot: UnicodePlot::DotCanvas
   }.freeze
 
   def self.included(mod)
@@ -122,3 +124,15 @@ class AsciiCanvasTest < Test::Unit::TestCase
 
   include CanvasTestCases
 end
+
+class DensityCanvasTest < Test::Unit::TestCase
+  CANVAS_NAME = :density
+
+  include CanvasTestCases
+end
+
+class DotCanvasTest < Test::Unit::TestCase
+  CANVAS_NAME = :dot
+
+  include CanvasTestCases
+end

data/test/test-densityplot.rb

@@ -0,0 +1,38 @@
+class DensityplotTest < Test::Unit::TestCase
+  include Helper::Fixture
+  include Helper::WithTerm
+
+  def setup
+    randn = fixture_path("randn_1338_2000.txt").read.each_line.map(&:to_f)
+    @dx = randn[0, 1000].compact
+    @dy = randn[1000, 1000].compact
+    assert_equal(1000, @dx.length)
+    assert_equal(1000, @dy.length)
+  end
+
+  test("default") do
+    plot = UnicodePlot.densityplot(@dx, @dy)
+    dx2 = @dx.map {|x| x + 2 }
+    dy2 = @dy.map {|y| y + 2 }
+    assert_same(plot,
+                UnicodePlot.densityplot!(plot, dx2, dy2))
+    _, output = with_term { plot.render($stdout) }
+    expected = fixture_path("scatterplot/densityplot.txt").read
+    assert_equal(output, expected)
+  end
+
+  test("parameters") do
+    plot = UnicodePlot.densityplot(@dx, @dy,
+                                   name: "foo",
+                                   color: :red,
+                                   title: "Title",
+                                   xlabel: "x")
+    dx2 = @dx.map {|x| x + 2 }
+    dy2 = @dy.map {|y| y + 2 }
+    assert_same(plot,
+                UnicodePlot.densityplot!(plot, dx2, dy2, name: "bar"))
+    _, output = with_term { plot.render($stdout) }
+    expected = fixture_path("scatterplot/densityplot_parameters.txt").read
+    assert_equal(output, expected)
+  end
+end

data/test/test-histogram.rb

@@ -0,0 +1,110 @@
+class HistogramTest < Test::Unit::TestCase
+  include Helper::Fixture
+  include Helper::WithTerm
+
+  sub_test_case("UnicodePlot.histogram") do
+    def setup
+      @x = fixture_path("randn.txt").read.lines.map(&:to_f)
+    end
+
+    test("default") do
+      plot = UnicodePlot.histogram(@x)
+      _, output = with_term { plot.render($stdout) }
+      assert_equal(fixture_path("histogram/default.txt").read,
+                   output)
+    end
+
+    test("nocolor") do
+      plot = UnicodePlot.histogram(@x)
+      output = StringIO.open do |sio|
+        plot.render(sio)
+        sio.close
+        sio.string
+      end
+      assert_equal(fixture_path("histogram/default_nocolor.txt").read,
+                   output)
+    end
+
+    test("losed: :left") do
+      plot = UnicodePlot.histogram(@x, closed: :left)
+      _, output = with_term { plot.render($stdout) }
+      assert_equal(fixture_path("histogram/default.txt").read,
+                   output)
+    end
+
+    test("x 100") do
+      x100 = @x.map {|a| a * 100 }
+      plot = UnicodePlot.histogram(x100)
+      _, output = with_term { plot.render($stdout) }
+      assert_equal(fixture_path("histogram/default_1e2.txt").read,
+                   output)
+    end
+
+    test("x0.01") do
+      x100 = @x.map {|a| a * 0.01 }
+      plot = UnicodePlot.histogram(x100)
+      _, output = with_term { plot.render($stdout) }
+      assert_equal(fixture_path("histogram/default_1e-2.txt").read,
+                   output)
+    end
+
+    test("xscale: :log10") do
+      plot = UnicodePlot.histogram(@x, xscale: :log10)
+      _, output = with_term { plot.render($stdout) }
+      assert_equal(fixture_path("histogram/log10.txt").read,
+                   output)
+    end
+
+    test("xscale: :log10 with custom label") do
+      plot = UnicodePlot.histogram(@x, xscale: :log10, xlabel: "custom label")
+      _, output = with_term { plot.render($stdout) }
+      assert_equal(fixture_path("histogram/log10_label.txt").read,
+                   output)
+    end
+
+    test("nbins: 5, closed: :right") do
+      plot = UnicodePlot.histogram(@x, nbins: 5, closed: :right)
+      _, output = with_term { plot.render($stdout) }
+      assert_equal(fixture_path("histogram/hist_params.txt").read,
+                   output)
+    end
+
+    test("with title, xlabel, color, margin, and padding") do
+      plot = UnicodePlot.histogram(@x,
+                                   title: "My Histogram",
+                                   xlabel: "Absolute Frequency",
+                                   color: :blue,
+                                   margin: 7,
+                                   padding: 3)
+      _, output = with_term { plot.render($stdout) }
+      assert_equal(fixture_path("histogram/parameters1.txt").read,
+                   output)
+    end
+
+    test("with title, xlabel, color, margin, padding, and labels: false") do
+      plot = UnicodePlot.histogram(@x,
+                                   title: "My Histogram",
+                                   xlabel: "Absolute Frequency",
+                                   color: :blue,
+                                   margin: 7,
+                                   padding: 3,
+                                   labels: false)
+      _, output = with_term { plot.render($stdout) }
+      assert_equal(fixture_path("histogram/parameters1_nolabels.txt").read,
+                   output)
+    end
+
+    test("with title, xlabel, color, border, symbol, and width") do
+      plot = UnicodePlot.histogram(@x,
+                                   title: "My Histogram",
+                                   xlabel: "Absolute Frequency",
+                                   color: :yellow,
+                                   border: :solid,
+                                   symbol: "=",
+                                   width: 50)
+      _, output = with_term { plot.render($stdout) }
+      assert_equal(fixture_path("histogram/parameters2.txt").read,
+                   output)
+    end
+  end
+end

data/test/test-lineplot.rb

@@ -125,6 +125,73 @@ class LineplotTest < Test::Unit::TestCase
                    output)
     end
 
+    test("limits") do
+      plot = UnicodePlot.lineplot(@x, @y, xlim: [-1.5, 3.5], ylim: [-5.5, 2.5])
+      _, output = with_term { plot.render($stdout) }
+      assert_equal(fixture_path("lineplot/limits.txt").read,
+                   output)
+    end
+
+    test("nogrid") do
+      plot = UnicodePlot.lineplot(@x, @y, grid: false)
+      _, output = with_term { plot.render($stdout) }
+      assert_equal(fixture_path("lineplot/nogrid.txt").read,
+                   output)
+    end
+
+    test("color: :blue") do
+      plot = UnicodePlot.lineplot(@x, @y, color: :blue, name: "points1")
+      _, output = with_term { plot.render($stdout) }
+      assert_equal(fixture_path("lineplot/blue.txt").read,
+                   output)
+    end
+
+    test("parameters") do
+      plot = UnicodePlot.lineplot(@x, @y,
+                                  name: "points1",
+                                  title: "Scatter",
+                                  xlabel: "x",
+                                  ylabel: "y")
+      _, output = with_term { plot.render($stdout) }
+      assert_equal(fixture_path("lineplot/parameters1.txt").read,
+                   output)
+
+      assert_same(plot,
+                  UnicodePlot.lineplot!(plot,
+                                        [0.5, 1, 1.5],
+                                        name: "points2"))
+      _, output = with_term { plot.render($stdout) }
+      assert_equal(fixture_path("lineplot/parameters2.txt").read,
+                   output)
+
+      assert_same(plot,
+                  UnicodePlot.lineplot!(plot,
+                                        [-0.5, 0.5, 1.5],
+                                        [0.5, 1, 1.5],
+                                        name: "points3"))
+      _, output = with_term { plot.render($stdout) }
+      assert_equal(fixture_path("lineplot/parameters3.txt").read,
+                   output)
+      output = StringIO.open do |sio|
+        plot.render(sio)
+        sio.close
+        sio.string
+      end
+      assert_equal(fixture_path("lineplot/nocolor.txt").read,
+                   output)
+    end
+
+    test("canvas size") do
+      plot = UnicodePlot.lineplot(@x, @y,
+                                  title: "Scatter",
+                                  canvas: :dot,
+                                  width: 10,
+                                  height: 5)
+      _, output = with_term { plot.render($stdout) }
+      assert_equal(fixture_path("lineplot/canvassize.txt").read,
+                   output)
+    end
+
     # TODO: functions
 
     # TODO: stairs

data/test/test-scatterplot.rb

@@ -0,0 +1,146 @@
+class ScatterplotTest < Test::Unit::TestCase
+  include Helper::Fixture
+  include Helper::WithTerm
+
+  def setup
+    @x = [-1, 1, 3, 3, -1]
+    @y = [2, 0, -5, 2, -5]
+  end
+
+  test("errors") do
+    assert_raise(ArgumentError) do
+      UnicodePlot.scatterplot()
+    end
+    assert_raise(ArgumentError) do
+      UnicodePlot.scatterplot([1, 2], [1, 2, 3])
+    end
+    assert_raise(ArgumentError) do
+      UnicodePlot.scatterplot([1, 2, 3], [1, 2])
+    end
+    assert_raise(ArgumentError) do
+      UnicodePlot.scatterplot(1..3, 1..2)
+    end
+  end
+
+  test("default") do
+    plot = UnicodePlot.scatterplot(@x, @y)
+    _, output = with_term { plot.render($stdout) }
+    assert_equal(fixture_path("scatterplot/default.txt").read,
+                 output)
+  end
+
+  test("y only") do
+    plot = UnicodePlot.scatterplot(@y)
+    _, output = with_term { plot.render($stdout) }
+    assert_equal(fixture_path("scatterplot/y_only.txt").read,
+                 output)
+  end
+
+  test("one range") do
+    plot = UnicodePlot.scatterplot(6..10)
+    _, output = with_term { plot.render($stdout) }
+    assert_equal(fixture_path("scatterplot/range1.txt").read,
+                 output)
+  end
+
+  test("two ranges") do
+    plot = UnicodePlot.scatterplot(11..15, 6..10)
+    _, output = with_term { plot.render($stdout) }
+    assert_equal(fixture_path("scatterplot/range2.txt").read,
+                 output)
+  end
+
+  test("scale1") do
+    x = @x.map {|a| a * 1e3 + 15 }
+    y = @y.map {|a| a * 1e-3 - 15 }
+    plot = UnicodePlot.scatterplot(x, y)
+    _, output = with_term { plot.render($stdout) }
+    assert_equal(fixture_path("scatterplot/scale1.txt").read,
+                 output)
+  end
+
+  test("scale2") do
+    x = @x.map {|a| a * 1e-3 + 15 }
+    y = @y.map {|a| a * 1e3 - 15 }
+    plot = UnicodePlot.scatterplot(x, y)
+    _, output = with_term { plot.render($stdout) }
+    assert_equal(fixture_path("scatterplot/scale2.txt").read,
+                 output)
+  end
+
+  test("scale3") do
+    miny = -1.2796649117521434e218
+    maxy = -miny
+    plot = UnicodePlot.scatterplot([1], [miny], xlim: [1, 1], ylim: [miny, maxy])
+    _, output = with_term { plot.render($stdout) }
+    expected = fixture_path("scatterplot/scale3.txt").read
+    assert_equal(expected, output)
+  end
+
+  test("limits") do
+    plot = UnicodePlot.scatterplot(@x, @y, xlim: [-1.5, 3.5], ylim: [-5.5, 2.5])
+    _, output = with_term { plot.render($stdout) }
+    assert_equal(fixture_path("scatterplot/limits.txt").read,
+                 output)
+  end
+
+  test("nogrid") do
+    plot = UnicodePlot.scatterplot(@x, @y, grid: false)
+    _, output = with_term { plot.render($stdout) }
+    assert_equal(fixture_path("scatterplot/nogrid.txt").read,
+                 output)
+  end
+
+  test("blue") do
+    plot = UnicodePlot.scatterplot(@x, @y, color: :blue, name: "points1")
+    _, output = with_term { plot.render($stdout) }
+    assert_equal(fixture_path("scatterplot/blue.txt").read,
+                 output)
+  end
+
+  test("parameters") do
+    plot = UnicodePlot.scatterplot(@x, @y,
+                                   name: "points1",
+                                   title: "Scatter",
+                                   xlabel: "x",
+                                   ylabel: "y")
+    _, output = with_term { plot.render($stdout) }
+    expected = fixture_path("scatterplot/parameters1.txt").read
+    assert_equal(expected, output)
+
+    assert_same(plot,
+                UnicodePlot.scatterplot!(plot,
+                                         [0.5, 1, 1.5],
+                                         name: "points2"))
+    _, output = with_term { plot.render($stdout) }
+    assert_equal(fixture_path("scatterplot/parameters2.txt").read,
+                 output)
+
+    assert_same(plot,
+                UnicodePlot.scatterplot!(plot,
+                                         [-0.5, 0.5, 1.5],
+                                         [0.5, 1, 1.5],
+                                         name: "points3"))
+    _, output = with_term { plot.render($stdout) }
+    assert_equal(fixture_path("scatterplot/parameters3.txt").read,
+                 output)
+    output = StringIO.open do |sio|
+      plot.render(sio)
+      sio.close
+      sio.string
+    end
+    assert_equal(fixture_path("scatterplot/nocolor.txt").read,
+                 output)
+  end
+
+  test("canvas size") do
+    plot = UnicodePlot.scatterplot(@x, @y,
+                                   title: "Scatter",
+                                   canvas: :dot,
+                                   width: 10,
+                                   height: 5)
+    _, output = with_term { plot.render($stdout) }
+    expected = fixture_path("scatterplot/canvassize.txt").read
+    assert_equal(expected, output)
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: unicode_plot
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - mrkn
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2019-06-14 00:00:00.000000000 Z
+date: 2019-06-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: enumerable-statistics
@@ -83,9 +83,16 @@ files:
 - lib/unicode_plot/boxplot.rb
 - lib/unicode_plot/braille_canvas.rb
 - lib/unicode_plot/canvas.rb
+- lib/unicode_plot/density_canvas.rb
+- lib/unicode_plot/densityplot.rb
+- lib/unicode_plot/dot_canvas.rb
+- lib/unicode_plot/grid_plot.rb
+- lib/unicode_plot/histogram.rb
 - lib/unicode_plot/lineplot.rb
+- lib/unicode_plot/lookup_canvas.rb
 - lib/unicode_plot/plot.rb
 - lib/unicode_plot/renderer.rb
+- lib/unicode_plot/scatterplot.rb
 - lib/unicode_plot/styled_printer.rb
 - lib/unicode_plot/utils.rb
 - lib/unicode_plot/value_transformer.rb
@@ -97,7 +104,10 @@ files:
 - test/test-barplot.rb
 - test/test-boxplot.rb
 - test/test-canvas.rb
+- test/test-densityplot.rb
+- test/test-histogram.rb
 - test/test-lineplot.rb
+- test/test-scatterplot.rb
 - unicode_plot.gemspec
 homepage: https://github.com/red-data-tools/unicode_plot.rb
 licenses:
@@ -123,11 +133,14 @@ signing_key:
 specification_version: 4
 summary: Plot your data by Unicode characters
 test_files:
+- test/test-scatterplot.rb
 - test/run-test.rb
 - test/test-barplot.rb
 - test/helper.rb
 - test/test-lineplot.rb
+- test/test-densityplot.rb
 - test/test-canvas.rb
+- test/test-histogram.rb
 - test/helper/with_term.rb
 - test/helper/fixture.rb
 - test/test-boxplot.rb