def test_betweenness_none():
drop_caches()
with settings.file_override("BETWEENNESS"):
with open(settings.BETWEENNESS, "wb") as f:
pickle.dump({"dog": 2, "cat": 3}, f)
assert set(SubstitutionFeaturesMixin._betweenness()) == {"dog", "cat"}
def test_betweenness_none_with_computed():
drop_caches()
# Lemmas are all lowercase.
for word in SubstitutionFeaturesMixin._betweenness():
assert word.islower() or is_int(word[0]) or is_int(word[-1]) or word in ["%", "!"]
def test_betweenness():
drop_caches()
assert SubstitutionFeaturesMixin._betweenness("dog") == 0.0046938277117769605
assert SubstitutionFeaturesMixin._betweenness("play") == 0.008277234906313704
assert np.isnan(SubstitutionFeaturesMixin._betweenness("wickiup"))
