v0.2.2

lib/uplot/command.rb

@@ -85,7 +85,7 @@ module Uplot
           params.padding = v
         end
            .on('-c', '--color VAL', String) do |v|
-          params.color = v.to_sym
+          params.color = v =~ /\A[0-9]+\z/ ? v.to_i : v.to_sym
         end
            .on('-x', '--xlabel VAL', String) do |v|
           params.xlabel = v

lib/uplot/preprocessing.rb

@@ -5,13 +5,32 @@ module Uplot
     module_function
 
     def input(input, delimiter, headers, transpose)
-      arr = read_csv(input, delimiter)
+      arr = parse_as_csv(input, delimiter)
       headers = get_headers(arr, headers, transpose)
       series = get_series(arr, headers, transpose)
-      Data.new(headers, series)
+      if headers.nil?
+        Data.new(headers, series)
+      else
+        if headers.include?(nil)
+          warn "\e[35mHeaders contains nil in it.\e[0m"
+        elsif headers.include? ''
+          warn "\e[35mHeaders contains \"\" in it.\e[0m"
+        end
+        h_size = headers.size
+        s_size = series.size
+        if h_size == s_size
+          Data.new(headers, series)
+        elsif h_size > s_size
+          warn "\e[35mThe number of headers is greater than the number of series.\e[0m"
+          exit 1
+        elsif h_size < s_size
+          warn "\e[35mThe number of headers is less than the number of series.\e[0m"
+          exit 1
+        end
+      end
     end
 
-    def read_csv(input, delimiter)
+    def parse_as_csv(input, delimiter)
       CSV.parse(input, col_sep: delimiter)
          .delete_if do |i|
            i == [] or i.all? nil

lib/uplot/version.rb

@@ -1,3 +1,3 @@
 module Uplot
-  VERSION = '0.2.1'.freeze
+  VERSION = '0.2.2'.freeze
 end

test/test_helper.rb

@@ -1,4 +1,6 @@
-$LOAD_PATH.unshift File.expand_path('../lib', __dir__)
+require 'simplecov'
+SimpleCov.start
+
 require 'uplot'
 
 require 'test/unit'

test/uplot/command_test.rb

@@ -0,0 +1,4 @@
+require_relative '../test_helper.rb'
+
+class UplotCommandTest < Test::Unit::TestCase
+end

test/uplot/plot_test.rb

@@ -0,0 +1,4 @@
+require_relative '../test_helper.rb'
+
+class UplotPlotTest < Test::Unit::TestCase
+end

test/uplot/preprocessing_test.rb

@@ -0,0 +1,130 @@
+require_relative '../test_helper.rb'
+
+class UplotPreprocessingTest < Test::Unit::TestCase
+  def setup
+    @m = Uplot::Preprocessing
+  end
+
+  test :transpose2 do
+    n = nil
+
+    assert_equal([[1, 2, 3],
+                  [4, 5, 6],
+                  [7, 8, 9]], @m.transpose2([[1, 4, 7],
+                                             [2, 5, 8],
+                                             [3, 6, 9]]))
+    assert_equal([[1, 2, 3],
+                  [4, 5, n],
+                  [6, n, n]], @m.transpose2([[1, 4, 6],
+                                             [2, 5],
+                                             [3]]))
+    assert_equal([[1, 2, 3],
+                  [n, 4, 5],
+                  [n, n, 6]], @m.transpose2([[1],
+                                             [2, 4],
+                                             [3, 5, 6]]))
+  end
+
+  test :get_headers do
+    assert_equal([1, 4, 7], @m.get_headers([[1, 2, 3],
+                                            [4, 5, 6],
+                                            [7, 8, 9]], true, true))
+
+    assert_equal([1, 2, 3], @m.get_headers([[1, 4, 6],
+                                            [2, 5],
+                                            [3]], true, true))
+
+    assert_equal([1, 2, 3], @m.get_headers([[1],
+                                            [2, 4],
+                                            [3, 5, 6]], true, true))
+
+    assert_equal([1, 2, 3], @m.get_headers([[1, 2, 3],
+                                            [4, 5, 6],
+                                            [7, 8, 9]], true, false))
+
+    assert_equal([1, 4, 6], @m.get_headers([[1, 4, 6],
+                                            [2, 5],
+                                            [3]], true, false))
+
+    assert_equal([1], @m.get_headers([[1],
+                                      [2, 4],
+                                      [3, 5, 6]], true, false))
+
+    assert_equal(nil, @m.get_headers([[1, 2, 3],
+                                      [4, 5, 6],
+                                      [7, 8, 9]], false, true))
+
+    assert_equal(nil, @m.get_headers([[1, 2, 3],
+                                      [4, 5, 6],
+                                      [7, 8, 9]], false, false))
+  end
+
+  test :get_series do
+    n = nil
+    
+    assert_equal([[2, 3], [5, 6], [8, 9]], @m.get_series([[1, 2, 3],
+                                                          [4, 5, 6],
+                                                          [7, 8, 9]], true, true))
+
+    assert_equal([[4, 6], [5], []], @m.get_series([[1, 4, 6],
+                                                   [2, 5],
+                                                   [3]], true, true))
+
+    assert_equal([[], [4], [5, 6]], @m.get_series([[1],
+                                                   [2, 4],
+                                                   [3, 5, 6]], true, true))
+
+    assert_equal([[4, 7], [5, 8], [6, 9]], @m.get_series([[1, 2, 3],
+                                                          [4, 5, 6],
+                                                          [7, 8, 9]], true, false))
+
+    assert_equal([[2, 3], [5, nil]], @m.get_series([[1, 4, 6],
+                                                    [2, 5],
+                                                    [3]], true, false))
+
+    assert_equal([[2, 3], [4, 5], [nil, 6]], @m.get_series([[1],
+                                                            [2, 4],
+                                                            [3, 5, 6]], true, false))
+
+    assert_equal([[1, 2, 3],
+                  [4, 5, 6],
+                  [7, 8, 9]], @m.get_series([[1, 2, 3],
+                                             [4, 5, 6],
+                                             [7, 8, 9]], false, true))
+
+    assert_equal([[1, 4, 6],
+                  [2, 5],
+                  [3]], @m.get_series([[1, 4, 6],
+                                       [2, 5],
+                                       [3]], false, true))
+
+    assert_equal([[1],
+                  [2, 4],
+                  [3, 5, 6]], @m.get_series([[1],
+                                             [2, 4],
+                                             [3, 5, 6]], false, true))
+
+    assert_equal([[1, 4, 7],
+                  [2, 5, 8],
+                  [3, 6, 9]], @m.get_series([[1, 2, 3],
+                                             [4, 5, 6],
+                                             [7, 8, 9]], false, false))
+
+    assert_equal([[1, 2, 3],
+                  [4, 5, n],
+                  [6, n, n]], @m.get_series([[1, 4, 6],
+                                             [2, 5],
+                                             [3]], false, false))
+
+    assert_equal([[1, 2, 3],
+                  [n, 4, 5],
+                  [n, n, 6]], @m.get_series([[1],
+                                             [2, 4],
+                                             [3, 5, 6]], false, false))
+  end
+
+  test :count do
+    assert_equal([%i[a b c], [3, 2, 1]], @m.count(%i[a a a b b c]))
+    assert_equal([%i[c b a], [3, 2, 1]], @m.count(%i[a b b c c c]))
+  end
+end

uplot.gemspec

@@ -24,5 +24,6 @@ Gem::Specification.new do |spec|
   spec.add_development_dependency 'bundler'
   spec.add_development_dependency 'rake'
   spec.add_development_dependency 'rubocop'
+  spec.add_development_dependency 'simplecov'
   spec.add_development_dependency 'test-unit'
 end