update tests with latest pyexcel api

Author: chfw

tests/test_database_io.py

@@ -26,7 +26,6 @@ def test_single_sheet_file(self):
                                      data={"file": (io, file_name)},
                                      content_type="multipart/form-data")
             ret = pe.get_array(file_type="xls", file_content=response.data)
-            print(ret)
             assert array == ret
 
 
@@ -60,7 +59,6 @@ def test_book_file(self):
                                      data={"file": (io, file_name)},
                                      content_type="multipart/form-data")
             ret = pe.get_book_dict(file_type="xls", file_content=response.data)
-            print(ret)
             assert data['Category'] == ret['category']
             sheet = pe.Sheet(data['Post'], name_columns_by_row=0)
             sheet.column.format("pub_date", lambda d: d.isoformat())

tests/test_make_responses.py

@@ -1,6 +1,6 @@
 from testapp import app
 import pyexcel as pe
-from _compact import OrderedDict, BytesIO
+from _compact import OrderedDict
 
 
 class TestSheet:
@@ -16,16 +16,14 @@ def setUp(self):
     def test_array(self):
         for struct_type in ["array", "dict", "records"]:
             print("Testing %s" % struct_type)
-            io = BytesIO()
-            sheet = pe.Sheet(self.data)
-            sheet.save_to_memory('xls', io)
+            io = pe.save_as(dest_file_type="xls", array=self.data)
             io.seek(0)
             response = self.app.post('/exchange/%s' % struct_type,
                                      buffered=True,
                                      data={"file": (io, "test.xls")},
                                      content_type="multipart/form-data")
             assert response.content_type == "application/vnd.ms-excel"
-            sheet = pe.load_from_memory('xls', response.data)
+            sheet = pe.get_sheet(file_type='xls', file_content=response.data)
             assert sheet.to_array() == self.data
 
 
@@ -41,14 +39,11 @@ def setUp(self):
 
     def test_book(self):
         for struct_type in ["book", "book_dict"]:
-            io = BytesIO()
-            book = pe.Book(self.content)
-            book.save_to_memory('xls', io)
-            io.seek(0)
+            io = pe.save_book_as(bookdict=self.content, dest_file_type="xls")
             response = self.app.post('/exchange/%s' % struct_type,
                                      buffered=True,
                                      data={"file": (io, "test.xls")},
                                      content_type="multipart/form-data")
             assert response.content_type == "application/vnd.ms-excel"
-            book2 = pe.load_book_from_memory('xls', response.data)
+            book2 = pe.get_book(file_type='xls', file_content=response.data)
             assert book2.to_dict() == self.content

tests/test_upload_n_download_excel.py

@@ -29,10 +29,7 @@ def test_download(self):
             file_name = 'test.%s' % upload_file_type
             for download_file_type in FILE_TYPE_MIME_TABLE.keys():
                 print("Uploading %s Downloading %s" % (upload_file_type, download_file_type))
-                io = pe.get_io(upload_file_type)
-                sheet = pe.Sheet(self.data)
-                sheet.save_to_memory(upload_file_type, io)
-                io.seek(0)
+                io = pe.save_as(dest_file_type=upload_file_type, array=self.data)
                 if not PY2:
                     if not isinstance(io, BytesIO):
                         io = BytesIO(io.getvalue().encode('utf-8'))
@@ -41,7 +38,8 @@ def test_download(self):
                                          data={"file": (io, file_name)},
                                          content_type="multipart/form-data")
                 assert response.content_type == FILE_TYPE_MIME_TABLE[download_file_type]
-                sheet = pe.load_from_memory(download_file_type, response.data)
+                sheet = pe.get_sheet(file_type=download_file_type,
+                                     file_content=response.data)
                 sheet.format(int)
                 array = sheet.to_array()
                 assert array == self.data