single sheet in book should be mentioned

pyexcel · May 12, 2015 · 9d8c57b · 9d8c57b
1 parent 8337f2d
commit 9d8c57b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_csv_book.py b/tests/test_csv_book.py
@@ -219,7 +219,7 @@ def setUp(self):
 
     def test_sheet_writer_to_memory(self):
         io = get_io(self.file_type)
-        w = CSVSheetWriter(io, None)
+        w = CSVSheetWriter(io, None, single_sheet_in_book=True)
         for row in self.data:
             w.write_row(row)
         w.close()
@@ -228,7 +228,7 @@ def test_sheet_writer_to_memory(self):
 
     def test_book_writer_to_memroy(self):
         io = get_io(self.file_type)
-        b = CSVWriter(io)
+        b = CSVWriter(io, single_sheet_in_book=True)
         w = b.create_sheet(None)
         for row in self.data:
             w.write_row(row)