Fix test

explosion · Oct 1, 2019 · fb943e6 · fb943e6
1 parent 851e6b1
commit fb943e6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/spacy_lookups_data/tests/test_lt.py b/spacy_lookups_data/tests/test_lt.py
@@ -25,4 +25,4 @@ def lt_lemmatizer():
 
 @pytest.mark.parametrize("tokens,lemmas", TEST_CASES)
 def test_lt_lemmatizer(lt_lemmatizer, tokens, lemmas):
-    assert lemmas == [lt_lemmatizer.lookup_table.get(token, token) for token in tokens]
+    assert lemmas == [lt_lemmatizer.lookup(token) for token in tokens]