☔ test #44

tests/test_new_csvz_book.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 import os
 from unittest import TestCase
 from pyexcel_io._compact import OrderedDict
@@ -17,13 +18,13 @@ class TestCSVZ(TestCase):
     file_type = 'csvz'
     writer_class = CSVZipBookWriter
     reader_class = CSVZipBookReader
-    result = "1,2,3"
+    result = u"中,文,1,2,3"
 
     def setUp(self):
         self.file = "csvz." + self.file_type
 
     def test_writing(self):
-        data = [[1, 2, 3]]
+        data = [[u'中', u'文', 1, 2, 3]]
         file_name = 'pyexcel_sheet1.' + self.file_type[0:3]
         zipbook = self.writer_class()
         zipbook.open(self.file)
@@ -32,13 +33,14 @@ class TestCSVZ(TestCase):
         zip = zipfile.ZipFile(self.file, 'r')
         self.assertEqual(zip.namelist(), [file_name])
         content = zip.read(file_name)
-        if not PY2:
-            content = content.decode('utf-8')
-        self.assertEqual(content.replace('\r', '').strip('\n'), self.result)
+        content = content.decode('utf-8')
+        self.assertEqual(
+            content.replace('\r', '').strip('\n'),
+            self.result)
         zip.close()
 
     def test_reading(self):
-        data = [[1, 2, 3]]
+        data = [[u'中', u'文', 1, 2, 3]]
         zipbook = self.writer_class()
         zipbook.open(self.file)
         zipbook.write({None: data})
@@ -46,7 +48,8 @@ class TestCSVZ(TestCase):
         zipreader = self.reader_class()
         zipreader.open(self.file)
         data = zipreader.read_all()
-        self.assertEqual(list(data['pyexcel_sheet1']), [[1, 2, 3]])
+        self.assertEqual(
+            list(data['pyexcel_sheet1']), [[u'中', u'文', 1, 2, 3]])
         zipreader.close()
 
     def tearDown(self):
@@ -57,7 +60,7 @@ class TestTSVZ(TestCSVZ):
     file_type = 'tsvz'
     writer_class = TSVZipBookWriter
     reader_class = TSVZipBookReader
-    result = "1\t2\t3"
+    result = u"中\t文\t1\t2\t3"
 
 
 def test_reading_from_memory():