This commit is contained in:
Adriane Boyd 2023-02-07 15:50:34 +01:00
parent aac8e1ba99
commit 97dda41d5d

View File

@ -33,7 +33,7 @@ def test_vocab_lexeme_prob(word1, word2, prob1, prob2):
assert "lexeme_prob" not in vocab.lookups.tables
# table is added when a prob is set
vocab[word1].prob = -1
assert "lexeme_prob" not in vocab.lookups.tables
assert "lexeme_prob" in vocab.lookups.tables
assert word2 not in vocab.lookups.get_table("lexeme_prob")
vocab[word2].prob = prob2
assert vocab[word1].prob > vocab[word2].prob