Merge #6

domitry · Aug 1, 2014 · 1b51dc0 · 1b51dc0
2 parents cd12354 + 5360f6f
commit 1b51dc0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/lib/nyaplot/data.rb b/lib/nyaplot/data.rb
@@ -54,10 +54,8 @@ def self.from_csv(*args)
       yield csv if block_given?
 
       head = if opts[:headers]
-               csv.headers
-             else
-               csv.readline
-             end
+        csv.headers if opts[:headers]
+      end
 
       rows = []
       csv.each do |row|

diff --git a/spec/data_spec.rb b/spec/data_spec.rb
@@ -1,5 +1,4 @@
 require 'spec_helper'
-require 'pry'
 
 describe Nyaplot::DataFrame do
   context ".from_csv" do
@@ -17,7 +16,8 @@
       end
 
       expect(df.column_labels).to eq(["image_resolution", "true_transform", "mls"])
+      expect(df["image_resolution"].first).to eq(6.55779)
       expect(df.column('true_transform').first[15]).to eq(1.0)
     end
   end
-end
+end