zhangyiss/YouPlot
1
0
Fork 0
mirror of https://github.com/red-data-tools/YouPlot.git synced 2025-05-07 15:41:12 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Allow DSV to accept single line header

Browse Source
This commit is contained in:
kojix2 2020-12-21 21:29:39 +09:00
parent 60fb611160
commit 6d707533a0
2 changed files with 17 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
lib/youplot/dsv.rb
View File

@ -57,17 +57,23 @@ module YouPlot
end end
def get_series(arr, headers, transpose) def get_series(arr, headers, transpose)
if transpose
if headers if headers
if arr.size > 1
if transpose
arr.map { |row| row[1..-1] } arr.map { |row| row[1..-1] }
else else
arr
end
elsif headers
transpose2(arr[1..-1]) transpose2(arr[1..-1])
end
else
Array.new(arr[0].size, [])
end
else
if transpose
arr
else else
transpose2(arr) transpose2(arr)
end end
end end
end end
end end
end

4
test/youplot/dev_reader_test.rb → test/youplot/dsv_test.rb
View File

@ -59,6 +59,8 @@ class YouPlotDSVTest < Test::Unit::TestCase
assert_equal(nil, @m.get_headers([[1, 2, 3], assert_equal(nil, @m.get_headers([[1, 2, 3],
[4, 5, 6], [4, 5, 6],
[7, 8, 9]], false, false)) [7, 8, 9]], false, false))
assert_equal([1, 2, 3], @m.get_headers([[1, 2, 3]], true, false))
end end
test :get_series do test :get_series do
@ -123,5 +125,7 @@ class YouPlotDSVTest < Test::Unit::TestCase
[n, n, 6]], @m.get_series([[1], [n, n, 6]], @m.get_series([[1],
[2, 4], [2, 4],
[3, 5, 6]], false, false)) [3, 5, 6]], false, false))
assert_equal([[], [] ,[]], @m.get_series([[1, 2, 3]], true, false))
end end
end end