def test_extrair_entidades_do_form(self):
ent1 = criar_entidade(slug='a')
ent2 = criar_entidade(slug='b')
post = {
'titulo': "super herois da liga justiça",
'entidades': '%s,%s' % (ent1.id, ent2.id),
}
request = HttpRequest()
request.POST = post
documento = extrair_dados_do_form(request)
assert documento['entidades'] == [ent1, ent2]
def test_extract_entity_mais_usada_ngrams(self):
e1 = criar_entidade(slug='ipad-3gs', nome="Ipad 3gs")
e2 = criar_entidade(slug='ipad', nome="Ipad")
e3 = criar_entidade(slug='sony', nome="Sony")
text = '<p>ipad 3gs é o tablet sony do sucesso ipad 3gs</p>'
assert e1 == extract_entity(text)
def test_extract_entity_mais_usada(self):
e1 = criar_entidade(slug='ipad', nome="Ipad")
e2 = criar_entidade(slug='sony', nome="Sony")
text = '<p>ipad é o tablet sony do sucesso sony</p>'
assert e2 == extract_entity(text)
def test_extract_entity(self):
e = criar_entidade(slug='ipad', nome="Ipad")
text = '<p>ipad é o tablet do sucesso</p>'
assert e == extract_entity(text)