diff --git a/spacy/tests/tagger/test_lemmatizer.py b/spacy/tests/tagger/test_lemmatizer.py index e90d62c84..8ab4422a3 100644 --- a/spacy/tests/tagger/test_lemmatizer.py +++ b/spacy/tests/tagger/test_lemmatizer.py @@ -58,7 +58,7 @@ def test_noun_lemmas(lemmatizer): def test_base_form_dive(lemmatizer): if lemmatizer is None: return None - + do = lemmatizer.noun assert do('dive', number='sing') == set(['dive']) assert do('dive', number='plur') == set(['diva']) @@ -67,7 +67,7 @@ def test_base_form_dive(lemmatizer): def test_base_form_saw(lemmatizer): if lemmatizer is None: return None - + do = lemmatizer.verb assert do('saw', verbform='past') == set(['see']) @@ -75,7 +75,7 @@ def test_base_form_saw(lemmatizer): def test_smart_quotes(lemmatizer): if lemmatizer is None: return None - + do = lemmatizer.punct assert do('“') == set(['"']) assert do('“') == set(['"']) @@ -84,10 +84,10 @@ def test_smart_quotes(lemmatizer): def test_pickle_lemmatizer(lemmatizer): if lemmatizer is None: return None - + file_ = io.BytesIO() pickle.dump(lemmatizer, file_) file_.seek(0) - + loaded = pickle.load(file_)