Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Next Next commit
take title for plot from Daru::DataFrame
  • Loading branch information
ievgrafov committed Jan 21, 2017
commit c0794d81b4e1e39f10f4b510e718e4cae3251878
5 changes: 5 additions & 0 deletions lib/gnuplotrb/plot.rb
Original file line number Diff line number Diff line change
Expand Up @@ -284,6 +284,7 @@ def parse_datasets_array(datasets)
when Hamster::Vector
datasets[0]
when (defined?(Daru) ? Daru::DataFrame : nil)
set_name_from_daru_dataframe(datasets[0])
Hamster::Vector.new(datasets[0].map { |ds| dataset_from_any(ds) })
else
Hamster::Vector.new(datasets.map { |ds| dataset_from_any(ds) })
Expand All @@ -295,5 +296,9 @@ def parse_datasets_array(datasets)
def new_with_options(options)
self.class.new(@datasets, options)
end

def set_name_from_daru_dataframe(dataframe)
self.title = dataframe.name
end
end
end
33 changes: 24 additions & 9 deletions spec/plot_spec.rb
Original file line number Diff line number Diff line change
Expand Up @@ -16,11 +16,6 @@
@title = 'Awesome spec'
@formula = %w(sin(x) cos(x) exp(-x))
@options = { title: @title, term: 'dumb' }
@df = Daru::DataFrame.new(
Build: [312, 630, 315, 312],
Test: [525, 1050, 701, 514],
Deploy: [215, 441, 370, 220]
)
end

it 'should be created out of sequence of datasets' do
Expand All @@ -38,10 +33,30 @@
expect(plot.title).to eql(@title)
end

it 'should be created out of Daru::DataFrame' do
p = Plot.new(@df)
expect(p).to be_an_instance_of(Plot)
expect(p.datasets.size).to be_eql(3)
context 'creation out of Daru::DataFrame' do
let(:df) do
Daru::DataFrame.new(
{
Build: [312, 630, 315, 312],
Test: [525, 1050, 701, 514],
Deploy: [215, 441, 370, 220]
},
index: [0, 1, 2, 3],
name: 'Dummy DataFrame'
)
end

subject { Plot.new(df) }

it { is_expected.to be_an_instance_of Plot }

it 'creates datasets for each column except index' do
expect(subject.datasets.size).to be 3
end

it 'takes name for plot from DataFrame' do
expect(subject.title).to eql(df.name)
end
end
end

Expand Down