fix fuck test

Author: alingse

.flake8

@@ -0,0 +1,2 @@
+[flake8]
+ignore = E501

jsoncsv/jsontool.py

@@ -2,6 +2,7 @@
 # author@alingse
 # 2016.05.27
 from __future__ import absolute_import
+from __future__ import unicode_literals
 
 import json
 
@@ -150,4 +151,4 @@ def convert_json(fin, fout, type="expand", separator=".", safe=False):
             fout.write(content)
         else:
             fout.write(content.encode('utf-8'))
-        fout.write('\n')
+        fout.write(str('\n'))

tests/test_jsoncsv.py

@@ -14,4 +14,5 @@ def test_jsoncsv_expand(self):
         runner = CliRunner()
         args = ['-e', 'fixture/files/raw.0.json', 'fixture/files/tmp.expand.0.json']
         result = runner.invoke(jsoncsv, args=args)
+        print(result)
         assert result.exit_code == 0

tests/test_jsontool.py

@@ -1,10 +1,12 @@
 # coding=utf-8
 # author@alingse
 # 2016.08.09
+from __future__ import unicode_literals
 
 import io
 import unittest
 
+from jsoncsv import PY3
 from jsoncsv.jsontool import expand, restore
 from jsoncsv.jsontool import is_array
 from jsoncsv.jsontool import convert_json
@@ -92,8 +94,11 @@ def test_expand_with_safe(self):
 class TestConvertJSON(unittest.TestCase):
 
     def test_convert_expand(self):
-        fin = io.BytesIO('{"a":{"b":3}}\n{"a":{"c":4}}\n')
-        fout = io.BytesIO()
+        fin = io.StringIO('{"a":{"b":3}}\n{"a":{"c":4}}\n')
+        if PY3:
+            fout = io.StringIO()
+        else:
+            fout = io.BytesIO()
 
         convert_json(fin, fout)
 
@@ -103,8 +108,11 @@ def test_convert_expand(self):
         fout.close()
 
     def test_convert_restore(self):
-        fin = io.BytesIO('{"a.b": 3}\n{"a.c": 4}\n')
-        fout = io.BytesIO()
+        fin = io.StringIO('{"a.b": 3}\n{"a.c": 4}\n')
+        if PY3:
+            fout = io.StringIO()
+        else:
+            fout = io.BytesIO()
 
         convert_json(fin, fout, type="restore")