mirror of
https://github.com/red-data-tools/YouPlot.git
synced 2025-09-19 02:18:08 +08:00
Compare commits
34 Commits
Author | SHA1 | Date | |
---|---|---|---|
![]() |
731daef3f8 | ||
![]() |
78363cd198 | ||
![]() |
0389f7fc5c | ||
![]() |
a33b0e7628 | ||
![]() |
4d62acea75 | ||
![]() |
e22976c1a2 | ||
![]() |
385d02d232 | ||
![]() |
0ce394a11d | ||
![]() |
3baada320e | ||
![]() |
b6c3ca9b43 | ||
![]() |
975eb95f55 | ||
![]() |
bd16c30613 | ||
![]() |
3a8c1e62f3 | ||
![]() |
522a111aa9 | ||
![]() |
7020785818 | ||
![]() |
c25d47d721 | ||
![]() |
40c5d9fbdf | ||
![]() |
3e3b1cdfec | ||
![]() |
661e5048dd | ||
![]() |
3b8846efbe | ||
![]() |
50cb8d7463 | ||
![]() |
b1df2ed544 | ||
![]() |
05c3b14acd | ||
![]() |
9e2f169e6c | ||
![]() |
d6b6ae963d | ||
![]() |
3492f52df9 | ||
![]() |
21626122eb | ||
![]() |
38084c6db5 | ||
![]() |
59d200cae0 | ||
![]() |
cd8ee7f9d9 | ||
![]() |
9144207c85 | ||
![]() |
e0914beec8 | ||
![]() |
e146bc66f3 | ||
![]() |
bc0204af53 |
2
Gemfile
2
Gemfile
@@ -1,3 +1,5 @@
|
||||
# frozen_string_literal: true
|
||||
|
||||
source 'https://rubygems.org'
|
||||
|
||||
# Specify your gem's dependencies in uplot.gemspec
|
||||
|
12
README.md
12
README.md
@@ -2,6 +2,8 @@
|
||||
|
||||
[](https://travis-ci.com/kojix2/uplot)
|
||||
[](https://badge.fury.io/rb/u-plot)
|
||||
[](https://rubydoc.info/gems/u-plot)
|
||||
[](LICENSE.txt)
|
||||
|
||||
Create ASCII charts on the terminal with data from standard streams in the pipeline.
|
||||
|
||||
@@ -17,7 +19,7 @@ gem install u-plot
|
||||
|
||||
## Usage
|
||||
|
||||
### histogram
|
||||
**histogram**
|
||||
|
||||
```sh
|
||||
ruby -r numo/narray -e "puts Numo::DFloat.new(1000).rand_norm.to_a" \
|
||||
@@ -46,6 +48,14 @@ ruby -r numo/narray -e "puts Numo::DFloat.new(1000).rand_norm.to_a" \
|
||||
Frequency
|
||||
```
|
||||
|
||||
**scatter**
|
||||
|
||||
```sh
|
||||
wget https://raw.githubusercontent.com/uiuc-cse/data-fa14/gh-pages/data/iris.csv -qO - \
|
||||
| cut -f1-4 -d, \
|
||||
| uplot scatter -H -d, -t IRIS -m 10
|
||||
```
|
||||
|
||||
## Development
|
||||
|
||||
## Contributing
|
||||
|
2
Rakefile
2
Rakefile
@@ -1,3 +1,5 @@
|
||||
# frozen_string_literal: true
|
||||
|
||||
require 'bundler/gem_tasks'
|
||||
require 'rake/testtask'
|
||||
|
||||
|
@@ -1,4 +1,5 @@
|
||||
#!/usr/bin/env ruby
|
||||
# frozen_string_literal: true
|
||||
|
||||
require 'uplot'
|
||||
|
||||
|
@@ -1,3 +1,5 @@
|
||||
# frozen_string_literal: true
|
||||
|
||||
require 'unicode_plot'
|
||||
require 'uplot/version'
|
||||
require 'uplot/preprocessing'
|
||||
|
@@ -1,254 +1,47 @@
|
||||
require 'optparse'
|
||||
# frozen_string_literal: true
|
||||
|
||||
require_relative 'preprocessing'
|
||||
require_relative 'command/parser'
|
||||
|
||||
module Uplot
|
||||
Data = Struct.new(:headers, :series)
|
||||
|
||||
class Command
|
||||
Params = Struct.new(
|
||||
# Sort me!
|
||||
:title,
|
||||
:width,
|
||||
:height,
|
||||
:border,
|
||||
:margin,
|
||||
:padding,
|
||||
:color,
|
||||
:xlabel,
|
||||
:ylabel,
|
||||
:labels,
|
||||
:symbol,
|
||||
:xscale,
|
||||
:nbins,
|
||||
:closed,
|
||||
:canvas,
|
||||
:xlim,
|
||||
:ylim,
|
||||
:grid,
|
||||
:name
|
||||
) do
|
||||
def to_hc
|
||||
to_h.compact
|
||||
end
|
||||
end
|
||||
|
||||
attr_accessor :params, :plot_type
|
||||
attr_reader :raw_inputs, :data, :fmt
|
||||
attr_accessor :params
|
||||
attr_reader :raw_inputs, :data, :fmt, :parser
|
||||
|
||||
def initialize
|
||||
@params = Params.new
|
||||
|
||||
@plot_type = nil
|
||||
@headers = nil
|
||||
@delimiter = "\t"
|
||||
@transpose = false
|
||||
@output = false
|
||||
@count = false
|
||||
@fmt = 'xyy'
|
||||
|
||||
@raw_inputs = []
|
||||
@debug = false
|
||||
end
|
||||
|
||||
def create_parser
|
||||
OptionParser.new do |opt|
|
||||
opt.program_name = 'uplot'
|
||||
opt.version = Uplot::VERSION
|
||||
opt.on('-o', '--output', TrueClass) do |v|
|
||||
@output = v
|
||||
end
|
||||
.on('-d', '--delimiter VAL', String) do |v|
|
||||
@delimiter = v
|
||||
end
|
||||
.on('-H', '--headers', TrueClass) do |v|
|
||||
@headers = v
|
||||
end
|
||||
.on('-T', '--transpose', TrueClass) do |v|
|
||||
@transpose = v
|
||||
end
|
||||
.on('-t', '--title VAL', String) do |v|
|
||||
params.title = v
|
||||
end
|
||||
.on('-w', '--width VAL', Numeric) do |v|
|
||||
params.width = v
|
||||
end
|
||||
.on('-h', '--height VAL', Numeric) do |v|
|
||||
params.height = v
|
||||
end
|
||||
.on('-b', '--border VAL', Numeric) do |v|
|
||||
params.border = v
|
||||
end
|
||||
.on('-m', '--margin VAL', Numeric) do |v|
|
||||
params.margin = v
|
||||
end
|
||||
.on('-p', '--padding VAL', Numeric) do |v|
|
||||
params.padding = v
|
||||
end
|
||||
.on('-c', '--color VAL', String) do |v|
|
||||
params.color = v.to_sym
|
||||
end
|
||||
.on('-x', '--xlabel VAL', String) do |v|
|
||||
params.xlabel = v
|
||||
end
|
||||
.on('-y', '--ylabel VAL', String) do |v|
|
||||
params.ylabel = v
|
||||
end
|
||||
.on('-l', '--labels', TrueClass) do |v|
|
||||
params.labels = v
|
||||
end
|
||||
.on('--fmt VAL', String) do |v|
|
||||
@fmt = v
|
||||
end
|
||||
.on('--debug', TrueClass) do |v|
|
||||
@debug = v
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
def parse_options(argv = ARGV)
|
||||
main_parser = create_parser
|
||||
parsers = Hash.new { |h, k| h[k] = create_parser }
|
||||
|
||||
parsers[:barplot] = \
|
||||
parsers[:bar]
|
||||
.on('--symbol VAL', String) do |v|
|
||||
params.symbol = v
|
||||
end
|
||||
.on('--xscale VAL', String) do |v|
|
||||
params.xscale = v
|
||||
end
|
||||
.on('--count', TrueClass) do |v|
|
||||
@count = v
|
||||
end
|
||||
|
||||
parsers[:count] = \
|
||||
parsers[:c] # barplot -c
|
||||
.on('--symbol VAL', String) do |v|
|
||||
params.symbol = v
|
||||
end
|
||||
|
||||
parsers[:histogram] = \
|
||||
parsers[:hist]
|
||||
.on('-n', '--nbins VAL', Numeric) do |v|
|
||||
params.nbins = v
|
||||
end
|
||||
.on('--closed VAL', String) do |v|
|
||||
params.closed = v
|
||||
end
|
||||
.on('--symbol VAL', String) do |v|
|
||||
params.symbol = v
|
||||
end
|
||||
|
||||
parsers[:lineplot] = \
|
||||
parsers[:line]
|
||||
.on('--canvas VAL', String) do |v|
|
||||
params.canvas = v
|
||||
end
|
||||
.on('--xlim VAL', String) do |v|
|
||||
params.xlim = get_lim(v)
|
||||
end
|
||||
.on('--ylim VAL', String) do |v|
|
||||
params.ylim = get_lim(v)
|
||||
end
|
||||
|
||||
parsers[:lineplots] = \
|
||||
parsers[:lines]
|
||||
.on('--canvas VAL', String) do |v|
|
||||
params.canvas = v
|
||||
end
|
||||
.on('--xlim VAL', String) do |v|
|
||||
params.xlim = get_lim(v)
|
||||
end
|
||||
.on('--ylim VAL', String) do |v|
|
||||
params.ylim = get_lim(v)
|
||||
end
|
||||
|
||||
parsers[:scatter] = \
|
||||
parsers[:s]
|
||||
.on('--canvas VAL', String) do |v|
|
||||
params.canvas = v
|
||||
end
|
||||
.on('--xlim VAL', String) do |v|
|
||||
params.xlim = get_lim(v)
|
||||
end
|
||||
.on('--ylim VAL', String) do |v|
|
||||
params.ylim = get_lim(v)
|
||||
end
|
||||
|
||||
parsers[:density] = \
|
||||
parsers[:d]
|
||||
.on('--grid', TrueClass) do |v|
|
||||
params.grid = v
|
||||
end
|
||||
.on('--xlim VAL', String) do |v|
|
||||
params.xlim = get_lim(v)
|
||||
end
|
||||
.on('--ylim VAL', String) do |v|
|
||||
params.ylim = get_lim(v)
|
||||
end
|
||||
|
||||
parsers[:boxplot] = \
|
||||
parsers[:box]
|
||||
.on('--xlim VAL', String) do |v|
|
||||
params.xlim = get_lim(v)
|
||||
end
|
||||
|
||||
# Preventing the generation of new sub-commands
|
||||
parsers.default = nil
|
||||
|
||||
# Usage and help messages
|
||||
main_parser.banner = \
|
||||
<<~MSG
|
||||
Program: uplot (Tools for plotting on the terminal)
|
||||
Version: #{Uplot::VERSION} (using unicode_plot #{UnicodePlot::VERSION})
|
||||
|
||||
Usage: uplot <command> [options]
|
||||
|
||||
Command: #{parsers.keys.join(' ')}
|
||||
|
||||
Options:
|
||||
MSG
|
||||
|
||||
begin
|
||||
main_parser.order!(argv)
|
||||
rescue OptionParser::ParseError => e
|
||||
warn "uplot: #{e.message}"
|
||||
exit 1
|
||||
end
|
||||
|
||||
@plot_type = argv.shift&.to_sym
|
||||
|
||||
unless parsers.has_key?(plot_type)
|
||||
if plot_type.nil?
|
||||
warn main_parser.help
|
||||
else
|
||||
warn "uplot: unrecognized command '#{plot_type}'"
|
||||
end
|
||||
exit 1
|
||||
end
|
||||
parser = parsers[plot_type]
|
||||
|
||||
begin
|
||||
parser.parse!(argv) unless argv.empty?
|
||||
rescue OptionParser::ParseError => e
|
||||
warn "uplot: #{e.message}"
|
||||
exit 1
|
||||
end
|
||||
end
|
||||
|
||||
def get_lim(str)
|
||||
str.split(/-|:|\.\./)[0..1].map(&:to_f)
|
||||
@parser = Parser.new
|
||||
end
|
||||
|
||||
def run
|
||||
parse_options
|
||||
parser.parse_options
|
||||
command = parser.command
|
||||
params = parser.params
|
||||
delimiter = parser.delimiter
|
||||
transpose = parser.transpose
|
||||
headers = parser.headers
|
||||
output = parser.output
|
||||
count = parser.count
|
||||
fmt = parser.fmt
|
||||
debug = parser.debug
|
||||
|
||||
if command == :colors
|
||||
Plot.colors
|
||||
exit
|
||||
end
|
||||
|
||||
# Sometimes the input file does not end with a newline code.
|
||||
while input = Kernel.gets(nil)
|
||||
input.freeze
|
||||
@raw_inputs << input
|
||||
@data = Preprocessing.input(input, @delimiter, @headers, @transpose)
|
||||
@data = Preprocessing.input(input, delimiter, headers, transpose)
|
||||
pp @data if @debug
|
||||
case plot_type
|
||||
case command
|
||||
when :bar, :barplot
|
||||
Plot.barplot(data, params, @count)
|
||||
when :count, :c
|
||||
@@ -259,17 +52,17 @@ module Uplot
|
||||
Plot.line(data, params)
|
||||
when :lines, :lineplots
|
||||
Plot.lines(data, params, fmt)
|
||||
when :scatter, :scatterplot
|
||||
when :scatter, :s
|
||||
Plot.scatter(data, params, fmt)
|
||||
when :density
|
||||
when :density, :d
|
||||
Plot.density(data, params, fmt)
|
||||
when :box, :boxplot
|
||||
Plot.boxplot(data, params)
|
||||
else
|
||||
raise "unrecognized plot_type: #{plot_type}"
|
||||
raise "unrecognized plot_type: #{command}"
|
||||
end.render($stderr)
|
||||
|
||||
print input if @output
|
||||
print input if output
|
||||
end
|
||||
end
|
||||
end
|
||||
|
32
lib/uplot/command/params.rb
Normal file
32
lib/uplot/command/params.rb
Normal file
@@ -0,0 +1,32 @@
|
||||
# frozen_string_literal: true
|
||||
|
||||
module Uplot
|
||||
class Command
|
||||
Params = Struct.new(
|
||||
# Sort me!
|
||||
:title,
|
||||
:width,
|
||||
:height,
|
||||
:border,
|
||||
:margin,
|
||||
:padding,
|
||||
:color,
|
||||
:xlabel,
|
||||
:ylabel,
|
||||
:labels,
|
||||
:symbol,
|
||||
:xscale,
|
||||
:nbins,
|
||||
:closed,
|
||||
:canvas,
|
||||
:xlim,
|
||||
:ylim,
|
||||
:grid,
|
||||
:name
|
||||
) do
|
||||
def to_hc
|
||||
to_h.compact
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
224
lib/uplot/command/parser.rb
Normal file
224
lib/uplot/command/parser.rb
Normal file
@@ -0,0 +1,224 @@
|
||||
# frozen_string_literal: true
|
||||
|
||||
require 'optparse'
|
||||
require_relative 'params'
|
||||
|
||||
module Uplot
|
||||
class Command
|
||||
class Parser
|
||||
attr_reader :command, :params,
|
||||
:delimiter, :transpose, :headers, :output, :count, :fmt, :debug
|
||||
|
||||
def initialize
|
||||
@command = nil
|
||||
@params = Params.new
|
||||
|
||||
@delimiter = "\t"
|
||||
@transpose = false
|
||||
@headers = nil
|
||||
@output = false
|
||||
@count = false
|
||||
@fmt = 'xyy'
|
||||
@debug = false
|
||||
end
|
||||
|
||||
def create_default_parser
|
||||
OptionParser.new do |opt|
|
||||
opt.program_name = 'uplot'
|
||||
opt.version = Uplot::VERSION
|
||||
opt.on('-O', '--output', TrueClass) do |v|
|
||||
@output = v
|
||||
end
|
||||
opt.on('-d', '--delimiter VAL', 'use DELIM instead of TAB for field delimiter', String) do |v|
|
||||
@delimiter = v
|
||||
end
|
||||
opt.on('-H', '--headers', 'specify that the input has header row', TrueClass) do |v|
|
||||
@headers = v
|
||||
end
|
||||
opt.on('-T', '--transpose', TrueClass) do |v|
|
||||
@transpose = v
|
||||
end
|
||||
opt.on('-t', '--title VAL', 'print string on the top of plot', String) do |v|
|
||||
params.title = v
|
||||
end
|
||||
opt.on('-x', '--xlabel VAL', 'print string on the bottom of the plot', String) do |v|
|
||||
params.xlabel = v
|
||||
end
|
||||
opt.on('-y', '--ylabel VAL', 'print string on the far left of the plot', String) do |v|
|
||||
params.ylabel = v
|
||||
end
|
||||
opt.on('-w', '--width VAL', 'number of characters per row', Integer) do |v|
|
||||
params.width = v
|
||||
end
|
||||
opt.on('-h', '--height VAL', 'number of rows', Numeric) do |v|
|
||||
params.height = v
|
||||
end
|
||||
opt.on('-b', '--border VAL', 'specify the style of the bounding box', String) do |v|
|
||||
params.border = v.to_sym
|
||||
end
|
||||
opt.on('-m', '--margin VAL', 'number of spaces to the left of the plot', Numeric) do |v|
|
||||
params.margin = v
|
||||
end
|
||||
opt.on('-p', '--padding VAL', 'space of the left and right of the plot', Numeric) do |v|
|
||||
params.padding = v
|
||||
end
|
||||
opt.on('-c', '--color VAL', 'color of the drawing', String) do |v|
|
||||
params.color = v =~ /\A[0-9]+\z/ ? v.to_i : v.to_sym
|
||||
end
|
||||
opt.on('--[no-]labels', 'hide the labels', TrueClass) do |v|
|
||||
params.labels = v
|
||||
end
|
||||
opt.on('--fmt VAL', 'xyy, xyxy', String) do |v|
|
||||
@fmt = v
|
||||
end
|
||||
opt.on('--debug', TrueClass) do |v|
|
||||
@debug = v
|
||||
end
|
||||
yield opt if block_given?
|
||||
end
|
||||
end
|
||||
|
||||
def main_parser
|
||||
@main_parser ||= create_default_parser do |main_parser|
|
||||
# Usage and help messages
|
||||
main_parser.banner = \
|
||||
<<~MSG
|
||||
Program: uplot (Tools for plotting on the terminal)
|
||||
Version: #{Uplot::VERSION} (using unicode_plot #{UnicodePlot::VERSION})
|
||||
|
||||
Usage: uplot <command> [options]
|
||||
|
||||
Command: barplot bar
|
||||
histogram hist
|
||||
lineplot line
|
||||
scatter s
|
||||
density d
|
||||
boxplot box
|
||||
colors
|
||||
|
||||
Options:
|
||||
MSG
|
||||
end
|
||||
end
|
||||
|
||||
def sub_parser
|
||||
@sub_parser ||= create_default_parser do |parser|
|
||||
parser.banner = <<~MSG
|
||||
Usage: uplot #{command} [options]
|
||||
|
||||
Options:
|
||||
MSG
|
||||
|
||||
case command
|
||||
when nil
|
||||
warn main_parser.help
|
||||
exit 1
|
||||
|
||||
when :barplot, :bar
|
||||
parser.on('--symbol VAL', String) do |v|
|
||||
params.symbol = v
|
||||
end
|
||||
parser.on('--xscale VAL', String) do |v|
|
||||
params.xscale = v
|
||||
end
|
||||
parser.on('--count', TrueClass) do |v|
|
||||
@count = v
|
||||
end
|
||||
|
||||
when :count, :c
|
||||
parser.on('--symbol VAL', String) do |v|
|
||||
params.symbol = v
|
||||
end
|
||||
|
||||
when :histogram, :hist
|
||||
parser.on('-n', '--nbins VAL', Numeric) do |v|
|
||||
params.nbins = v
|
||||
end
|
||||
parser.on('--closed VAL', String) do |v|
|
||||
params.closed = v
|
||||
end
|
||||
parser.on('--symbol VAL', String) do |v|
|
||||
params.symbol = v
|
||||
end
|
||||
|
||||
when :lineplot, :line
|
||||
parser.on('--canvas VAL', String) do |v|
|
||||
params.canvas = v
|
||||
end
|
||||
parser.on('--xlim VAL', Array) do |v|
|
||||
params.xlim = v.take(2)
|
||||
end
|
||||
parser.on('--ylim VAL', Array) do |v|
|
||||
params.ylim = v.take(2)
|
||||
end
|
||||
|
||||
when :lineplots, :lines
|
||||
parser.on('--canvas VAL', String) do |v|
|
||||
params.canvas = v
|
||||
end
|
||||
parser.on('--xlim VAL', Array) do |v|
|
||||
params.xlim = v.take(2)
|
||||
end
|
||||
parser.on('--ylim VAL', Array) do |v|
|
||||
params.ylim = v.take(2)
|
||||
end
|
||||
|
||||
when :scatter, :s
|
||||
parser.on('--canvas VAL', String) do |v|
|
||||
params.canvas = v
|
||||
end
|
||||
parser.on('--xlim VAL', Array) do |v|
|
||||
params.xlim = v.take(2)
|
||||
end
|
||||
parser.on('--ylim VAL', Array) do |v|
|
||||
params.ylim = v.take(2)
|
||||
end
|
||||
|
||||
when :density, :d
|
||||
parser.on('--grid', TrueClass) do |v|
|
||||
params.grid = v
|
||||
end
|
||||
parser.on('--xlim VAL', Array) do |v|
|
||||
params.xlim = v.take(2)
|
||||
end
|
||||
parser.on('--ylim VAL', Array) do |v|
|
||||
params.ylim = v.take(2)
|
||||
end
|
||||
|
||||
when :boxplot, :box
|
||||
parser.on('--xlim VAL', Array) do |v|
|
||||
params.xlim = v.take(2)
|
||||
end
|
||||
|
||||
when :colors
|
||||
parser.on('-n', '--names', TrueClass) do |v|
|
||||
@color_names = v
|
||||
end
|
||||
|
||||
else
|
||||
warn "uplot: unrecognized command '#{command}'"
|
||||
exit 1
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
def parse_options(argv = ARGV)
|
||||
begin
|
||||
main_parser.order!(argv)
|
||||
rescue OptionParser::ParseError => e
|
||||
warn "uplot: #{e.message}"
|
||||
exit 1
|
||||
end
|
||||
|
||||
@command = argv.shift&.to_sym
|
||||
|
||||
begin
|
||||
sub_parser.parse!(argv)
|
||||
rescue OptionParser::ParseError => e
|
||||
warn "uplot: #{e.message}"
|
||||
exit 1
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
@@ -1,3 +1,5 @@
|
||||
# frozen_string_literal: true
|
||||
|
||||
require 'unicode_plot'
|
||||
|
||||
module Uplot
|
||||
@@ -123,6 +125,18 @@ module Uplot
|
||||
UnicodePlot.boxplot(headers, series, **params.to_hc)
|
||||
end
|
||||
|
||||
def colors
|
||||
UnicodePlot::StyledPrinter::TEXT_COLORS.each do |k, v|
|
||||
print v
|
||||
print k
|
||||
print "\t"
|
||||
print ' ●'
|
||||
print "\033[0m"
|
||||
print "\t"
|
||||
end
|
||||
puts
|
||||
end
|
||||
|
||||
def check_series_size(data, fmt)
|
||||
series = data.series
|
||||
if series.size == 1
|
||||
|
@@ -1,3 +1,5 @@
|
||||
# frozen_string_literal: true
|
||||
|
||||
require 'csv'
|
||||
|
||||
module Uplot
|
||||
@@ -5,13 +7,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)
|
||||
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
|
||||
|
@@ -1,3 +1,5 @@
|
||||
# frozen_string_literal: true
|
||||
|
||||
module Uplot
|
||||
VERSION = '0.2.0'.freeze
|
||||
VERSION = '0.2.4'
|
||||
end
|
||||
|
@@ -1,4 +1,8 @@
|
||||
$LOAD_PATH.unshift File.expand_path('../lib', __dir__)
|
||||
# frozen_string_literal: true
|
||||
|
||||
require 'simplecov'
|
||||
SimpleCov.start
|
||||
|
||||
require 'uplot'
|
||||
|
||||
require 'test/unit'
|
||||
|
6
test/uplot/command_test.rb
Normal file
6
test/uplot/command_test.rb
Normal file
@@ -0,0 +1,6 @@
|
||||
# frozen_string_literal: true
|
||||
|
||||
require_relative '../test_helper.rb'
|
||||
|
||||
class UplotCommandTest < Test::Unit::TestCase
|
||||
end
|
6
test/uplot/plot_test.rb
Normal file
6
test/uplot/plot_test.rb
Normal file
@@ -0,0 +1,6 @@
|
||||
# frozen_string_literal: true
|
||||
|
||||
require_relative '../test_helper.rb'
|
||||
|
||||
class UplotPlotTest < Test::Unit::TestCase
|
||||
end
|
132
test/uplot/preprocessing_test.rb
Normal file
132
test/uplot/preprocessing_test.rb
Normal file
@@ -0,0 +1,132 @@
|
||||
# frozen_string_literal: true
|
||||
|
||||
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
|
@@ -1,3 +1,5 @@
|
||||
# frozen_string_literal: true
|
||||
|
||||
require 'test_helper'
|
||||
|
||||
class UplotTest < Test::Unit::TestCase
|
||||
|
@@ -1,3 +1,5 @@
|
||||
# frozen_string_literal: true
|
||||
|
||||
require_relative 'lib/uplot/version'
|
||||
|
||||
Gem::Specification.new do |spec|
|
||||
@@ -24,5 +26,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
|
||||
|
Reference in New Issue
Block a user