Rooba
/
jellyfish
mirror of https://github.com/jamesturk/jellyfish.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

pass encoding on py3

This commit is contained in:
James Turk 2015-04-23 00:41:31 -04:00
parent 99947079c5
commit 78236905e8
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
jellyfish/test.py
View File

@ -2,10 +2,11 @@
import sys
if sys.version_info[0] < 3:
import unicodecsv as csv
open_kwargs = {}
else:
import csv
open_kwargs = {'encoding': 'utf8'}
import platform
import codecs
import pytest
@ -29,7 +30,7 @@ def jf(request):
def _load_data(name):
with codecs.open('testdata/{}.csv'.format(name), encoding='utf8') as f:
with open('testdata/{}.csv'.format(name), **open_kwargs) as f:
for data in csv.reader(f):
yield data
@ -96,7 +97,7 @@ def test_match_rating_comparison(jf, s1, s2, value):
# assert jf.porter_stem(a) == b
def test_porter_stem(jf):
with codecs.open('testdata/porter.csv', encoding='utf8') as f:
with open('testdata/porter.csv', **open_kwargs) as f:
reader = csv.reader(f)
for (a, b) in reader:
assert jf.porter_stem(a) == b