def test_expand_terms(multi_ling_frame):
vectors = VectorSpaceWrapper(frame=multi_ling_frame)
vectors.load()
term = [('/c/en/ski_jumper', 1.0)]
expanded_terms = vectors.expand_terms(terms=term, oov_vector=True)
expected_expanded_terms = [
('/c/en/ski_jumper', 0.9900990099009901),
('/c/en/ski_jumping', 0.009900990099009901),
]
assert expected_expanded_terms == expanded_terms
def test_expand_terms():
vectors = VectorSpaceWrapper(frame=TEST_FRAME)
vectors.load()
term = [('/c/en/ski_jumper', 1.0)]
expanded_terms = vectors.expand_terms(terms=term, limit_per_term=2, oov_vector=True)
expected_expanded_terms = [
('/c/en/ski_jumper', 0.9523809523809523),
('/c/pt/saltadores_de_esqui', 0.019047619047619046),
('/c/pl/skoczek_narciarski', 0.019047619047619046),
('/c/en/ski_jumping', 0.009523809523809523),
]
eq_(expected_expanded_terms, expanded_terms)
def test_expand_terms():
vectors = VectorSpaceWrapper(frame=TEST_FRAME)
vectors.load()
term = [('/c/en/ski_jumper', 1.0)]
expanded_terms = vectors.expand_terms(terms=term,
limit_per_term=2,
oov_vector=True)
expected_expanded_terms = [
('/c/en/ski_jumper', 0.9523809523809523),
('/c/pt/saltadores_de_esqui', 0.019047619047619046),
('/c/pl/skoczek_narciarski', 0.019047619047619046),
('/c/en/ski_jumping', 0.009523809523809523),
]
eq_(expected_expanded_terms, expanded_terms)
def test_expand_terms():
vectors = VectorSpaceWrapper(frame=TEST_FRAME)
vectors.load()
term = [('/c/en/ski_jumper', 1.0)]
expanded_terms = vectors.expand_terms(terms=term,
limit_per_term=2,
oov_vector=True)
# /c/en/bounder and /c/en/skier from neighbor search
# /c/en/ski_jumping from prefix match
expected_expanded_terms = [('/c/en/ski_jumper', 0.9523809523809523),
('/c/en/bounder', 0.019047619047619046),
('/c/en/skier', 0.019047619047619046),
('/c/en/ski_jumping', 0.009523809523809523)]
eq_(expected_expanded_terms, expanded_terms)
