Add the Processing module (tentative)

This commit is contained in:
kojix2 2020-11-25 17:31:41 +09:00
parent 9849898cb1
commit f2bd99ed2e
3 changed files with 37 additions and 1 deletions

View File

@ -0,0 +1,24 @@
# frozen_string_literal: true
module YouPlot
# plotting functions.
module Backends
module Processing
module_function
def count_values(arr)
# tally was added in Ruby 2.7
if Enumerable.method_defined? :tally
arr.tally
else
# https://github.com/marcandre/backports
arr.each_with_object(Hash.new(0)) { |item, res| res[item] += 1 }
.tap { |h| h.default = nil }
end
.sort { |a, b| a[1] <=> b[1] }
.reverse
.transpose
end
end
end
end

View File

@ -1,5 +1,6 @@
# frozen_string_literal: true
require_relative 'processing'
require 'unicode_plot'
module YouPlot
@ -13,7 +14,7 @@ module YouPlot
series = data.series
# `uplot count`
if count
series = Preprocessing.count_values(series[0])
series = Processing.count_values(series[0])
params.title = headers[0] if headers
end
if series.size == 1

View File

@ -0,0 +1,11 @@
# frozen_string_literal: true
require_relative '../../test_helper'
class YouPlotCommandTest < Test::Unit::TestCase
test :count_values do
@m = YouPlot::Backends::Processing
assert_equal([%i[a b c], [3, 2, 1]], @m.count_values(%i[a a a b b c]))
assert_equal([%i[c b a], [3, 2, 1]], @m.count_values(%i[a b b c c c]))
end
end