From 2f2c132d250212e7daad4bfc3e153dd622310658 Mon Sep 17 00:00:00 2001 From: Yusuke Sangenya Date: Mon, 9 Jan 2017 01:09:25 +0900 Subject: [PATCH 1/2] create test --- spec/dataframe_spec.rb | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/spec/dataframe_spec.rb b/spec/dataframe_spec.rb index b5165cf24..07a357b4b 100644 --- a/spec/dataframe_spec.rb +++ b/spec/dataframe_spec.rb @@ -2594,6 +2594,13 @@ @df.vectors = Daru::Index.new([1,2,'3',4,'5']) }.to raise_error(ArgumentError) end + + it "change name of vectors in @data" do + new_index_array = [:k, :l, :m] + @df.vectors = Daru::Index.new(new_index_array) + + expect(@df.data.map { |vector| vector.name }).to eq(new_index_array) + end end context "#rename_vectors" do From 88fc8281ea7efd823c78dd4647ae36c447a38595 Mon Sep 17 00:00:00 2001 From: Yusuke Sangenya Date: Mon, 9 Jan 2017 01:11:13 +0900 Subject: [PATCH 2/2] DataFrame#vector= changes the name of vectors in @data --- lib/daru/dataframe.rb | 3 +++ 1 file changed, 3 insertions(+) diff --git a/lib/daru/dataframe.rb b/lib/daru/dataframe.rb index a67805406..5388cf61a 100644 --- a/lib/daru/dataframe.rb +++ b/lib/daru/dataframe.rb @@ -1356,6 +1356,9 @@ def vectors= new_index end @vectors = new_index + @data.zip(new_index.to_a).each do |vect, name| + vect.name = name + end self end