test/row_reader_test.rb in iostreams-0.16.0 vs test/row_reader_test.rb in iostreams-0.16.1

@@ -11,23 +11,25 @@
     end
 
     describe '.open' do
       it 'file' do
         rows = []
-        IOStreams::Row::Reader.open(file_name) do |io|
+        count = IOStreams::Row::Reader.open(file_name) do |io|
           io.each { |row| rows << row }
         end
         assert_equal expected, rows
+        assert_equal expected.size, count
       end
 
       it 'stream' do
         rows = []
-        IOStreams.line_reader(file_name) do |file|
+        count = IOStreams.line_reader(file_name) do |file|
           IOStreams::Row::Reader.open(file) do |io|
             io.each { |row| rows << row }
           end
         end
         assert_equal expected, rows
+        assert_equal expected.size, count
       end
     end
 
   end
 end