test/xlsx_reader_test.rb in iostreams-0.8.2 vs test/xlsx_reader_test.rb in iostreams-0.9.0

@@ -1,13 +1,15 @@
 require_relative 'test_helper'
 
 module Streams
   describe IOStreams::Xlsx::Reader do
-    XLSX_CONTENTS = [
-      "first column,second column,third column",
-      "data 1,data 2,more data",
-    ]
+    before do
+      @xlsx_contents = [
+        ['first column', 'second column', 'third column'],
+        ['data 1', 'data 2', 'more data']
+      ]
+    end
 
     describe '.open' do
       let(:file_name) { File.join(File.dirname(__FILE__), 'files', 'spreadsheet.xlsx') }
 
       describe 'with a file path' do
@@ -16,26 +18,26 @@
         end
 
         it 'returns the contents of the file' do
           rows = []
           IOStreams::Xlsx::Reader.open(@file) do |spreadsheet|
-            spreadsheet.each_line { |row| rows << row }
+            spreadsheet.each { |row| rows << row }
           end
-          assert_equal(XLSX_CONTENTS, rows)
+          assert_equal(@xlsx_contents, rows)
         end
       end
 
       describe 'with a file stream' do
 
         it 'returns the contents of the file' do
           rows = []
           File.open(file_name) do |file|
             IOStreams::Xlsx::Reader.open(file) do |spreadsheet|
-              spreadsheet.each_line { |row| rows << row }
+              spreadsheet.each { |row| rows << row }
             end
           end
 
-          assert_equal(XLSX_CONTENTS, rows)
+          assert_equal(@xlsx_contents, rows)
         end
       end
     end
 
   end