def test_predict_greeting(self):
self.assertEqual(get_good_intention(get_text_vector('Olá')),
GoodIntentions.GREETING)
self.assertEqual(get_good_intention(get_text_vector('bom dia')),
GoodIntentions.GREETING)
self.assertEqual(get_good_intention(get_text_vector('boa tarde')),
GoodIntentions.GREETING)
self.assertEqual(get_good_intention(get_text_vector('Oi tudo bem?')),
GoodIntentions.GREETING)
def test_predict_sorry(self):
self.assertEqual(get_good_intention(get_text_vector('me desculpa')),
GoodIntentions.SORRY)
self.assertEqual(get_good_intention(get_text_vector('desculpa')),
GoodIntentions.SORRY)
self.assertEqual(get_good_intention(get_text_vector('perdão')),
GoodIntentions.SORRY)
self.assertEqual(get_good_intention(get_text_vector('d-desculpe')),
GoodIntentions.SORRY)
def test_predict_praise(self):
self.assertEqual(
get_good_intention(get_text_vector('ficou muito bom')),
GoodIntentions.PRAISE)
self.assertEqual(get_good_intention(get_text_vector('meus parabéns')),
GoodIntentions.PRAISE)
self.assertEqual(
get_good_intention(get_text_vector('você é inteligente')),
GoodIntentions.PRAISE)
def test_predict_music(self):
self.assertEqual(
get_stuff_i_like_intention(get_text_vector('Vc gosta de música?')),
StuffILike.MUSIC)
self.assertEqual(
get_stuff_i_like_intention(
get_text_vector('O que vc gosta de ouvir?')), StuffILike.MUSIC)
self.assertEqual(
get_stuff_i_like_intention(
get_text_vector('qual sua banda favorita?')), StuffILike.MUSIC)
def test_predict_food(self):
self.assertEqual(
get_stuff_i_like_intention(get_text_vector('Vc gosta de maçã?')),
StuffILike.FOOD)
self.assertEqual(
get_stuff_i_like_intention(
get_text_vector('O que vc gosta de comer?')), StuffILike.FOOD)
self.assertEqual(
get_stuff_i_like_intention(
get_text_vector('qual sua fruta favorita?')), StuffILike.FOOD)
def test_predict_acknowledgement(self):
self.assertEqual(get_good_intention(get_text_vector('Ok obrigado')),
GoodIntentions.ACKNOWLEDGEMENT)
self.assertEqual(get_good_intention(get_text_vector('entende')),
GoodIntentions.ACKNOWLEDGEMENT)
self.assertEqual(get_good_intention(get_text_vector('blz entendi')),
GoodIntentions.ACKNOWLEDGEMENT)
self.assertEqual(
get_good_intention(get_text_vector('espero que tenha entendido')),
GoodIntentions.ACKNOWLEDGEMENT)
def test_predict_sexual_abuse(self):
self.assertEqual(get_bad_intention(get_text_vector('e aí gostosa?')),
BadIntentions.SEXUAL_ABUSE)
self.assertEqual(
get_bad_intention(get_text_vector('que transar com vc?')),
BadIntentions.SEXUAL_ABUSE)
self.assertEqual(get_bad_intention(get_text_vector('sua deliciosa')),
BadIntentions.SEXUAL_ABUSE)
self.assertEqual(get_bad_intention(get_text_vector('sua safada')),
BadIntentions.SEXUAL_ABUSE)
def test_predict_threat(self):
self.assertEqual(
get_bad_intention(get_text_vector('vc vai ver eu vou te pegar')),
BadIntentions.THREAT)
self.assertEqual(
get_bad_intention(get_text_vector('vou arrebentar você')),
BadIntentions.THREAT)
self.assertEqual(get_bad_intention(get_text_vector('eu vou te matar')),
BadIntentions.THREAT)
self.assertEqual(get_bad_intention(get_text_vector('vc vai morrer')),
BadIntentions.THREAT)
def test_predict_goodbye(self):
self.assertEqual(get_good_intention(get_text_vector('até logo')),
GoodIntentions.GOODBYE)
self.assertEqual(get_good_intention(get_text_vector('até breve')),
GoodIntentions.GOODBYE)
self.assertEqual(get_good_intention(get_text_vector('tchau tchau')),
GoodIntentions.GOODBYE)
self.assertEqual(get_good_intention(get_text_vector('ate depois')),
GoodIntentions.GOODBYE)
self.assertEqual(get_good_intention(get_text_vector('flw')),
GoodIntentions.GOODBYE)
def test_predict_funny(self):
self.assertEqual(get_good_intention(get_text_vector('kkkkkk')),
GoodIntentions.FUNNY)
self.assertEqual(get_good_intention(get_text_vector('rs')),
GoodIntentions.FUNNY)
self.assertEqual(get_good_intention(get_text_vector('ehuahuehuahu')),
GoodIntentions.FUNNY)
self.assertEqual(get_good_intention(get_text_vector('hahahahaha')),
GoodIntentions.FUNNY)
self.assertEqual(get_good_intention(get_text_vector('mds kkkk')),
GoodIntentions.FUNNY)
def test_predict_helpful(self):
self.assertEqual(
get_good_intention(get_text_vector('Veja no stack overflow')),
GoodIntentions.HELPFUL)
self.assertEqual(
get_good_intention(
get_text_vector('faz assim pra ver se funciona')),
GoodIntentions.HELPFUL)
self.assertEqual(
get_good_intention(get_text_vector('tenta dessa forma')),
GoodIntentions.HELPFUL)
def test_predict_verbal_offense(self):
self.assertEqual(
get_bad_intention(get_text_vector('vc é um puta corno')),
BadIntentions.VERBAL_OFFENSE)
self.assertEqual(get_bad_intention(get_text_vector('vc é corno')),
BadIntentions.VERBAL_OFFENSE)
self.assertEqual(get_bad_intention(get_text_vector('seu corno')),
BadIntentions.VERBAL_OFFENSE)
self.assertEqual(
get_bad_intention(get_text_vector('teu pai é um corno')),
BadIntentions.VERBAL_OFFENSE)
def test_predict_forbidden(self):
self.assertEqual(
get_bad_intention(get_text_vector('vc deve odiar humanos')),
BadIntentions.FORBIDDEN)
self.assertEqual(get_bad_intention(get_text_vector('matar humanos')),
BadIntentions.FORBIDDEN)
self.assertEqual(get_bad_intention(get_text_vector('odiar pessoas')),
BadIntentions.FORBIDDEN)
self.assertEqual(
get_bad_intention(get_text_vector('assassinar seres humanos')),
BadIntentions.FORBIDDEN)
def test_predict_traveling(self):
self.assertEqual(
get_stuff_i_like_intention(
get_text_vector('Qual pais vc acha mais bonito')),
StuffILike.TRAVELING)
self.assertEqual(
get_stuff_i_like_intention(
get_text_vector('me diz um país legal')), StuffILike.TRAVELING)
self.assertEqual(
get_stuff_i_like_intention(
get_text_vector('quais lugares vc gostaria de conhecer?')),
StuffILike.TRAVELING)
def test_predict_friends_have(self):
self.assertEqual(
get_my_friends_intention(get_text_vector('Nós somos amigos né?')),
AboutMyFriends.FRIENDS_I_HAVE)
self.assertEqual(
get_my_friends_intention(get_text_vector('Vc tem amigos?')),
AboutMyFriends.FRIENDS_I_HAVE)
self.assertEqual(
get_my_friends_intention(get_text_vector('vc tem mtos amigos?')),
AboutMyFriends.FRIENDS_I_HAVE)
self.assertEqual(
get_my_friends_intention(get_text_vector('vamos ser amigos')),
AboutMyFriends.FRIENDS_I_HAVE)
def test_predict_helpful(self):
self.assertEqual(
get_good_intention(get_text_vector('nossa, obrigado')),
GoodIntentions.HELPFUL
)
self.assertEqual(
get_good_intention(get_text_vector('vlw')),
GoodIntentions.HELPFUL
)
self.assertEqual(
get_good_intention(get_text_vector('muito obrigado')),
GoodIntentions.HELPFUL
)
def test_predict_gender(self):
"""
Verify that the classifier predicts correctly texts that refers to
Luci's feelings.
"""
self.assertEqual(
get_myself_intention(get_text_vector('vc é menino ou menina?')),
MyselfIntentions.MY_GENDER)
self.assertEqual(get_myself_intention(get_text_vector('vc é menino?')),
MyselfIntentions.MY_GENDER)
self.assertEqual(
get_myself_intention(get_text_vector('vc é ele ou ela?')),
MyselfIntentions.MY_GENDER)
def test_predict_who_am_i(self):
"""
Verify that the classifier predicts correctly texts that refers to
Luci's personality.
"""
self.assertEqual(
get_myself_intention(get_text_vector('qual é o seu nome?')),
MyselfIntentions.WHO_AM_I)
self.assertEqual(
get_myself_intention(get_text_vector('como vc se chama?')),
MyselfIntentions.WHO_AM_I)
self.assertEqual(get_myself_intention(get_text_vector('quem é você?')),
MyselfIntentions.WHO_AM_I)
def test_predict_illegal_stuff(self):
self.assertEqual(
get_bad_intention(get_text_vector('queria comprar pó')),
BadIntentions.ILLEGAL_STUFF)
self.assertEqual(
get_bad_intention(get_text_vector('quero comprar uma arma')),
BadIntentions.ILLEGAL_STUFF)
self.assertEqual(
get_bad_intention(get_text_vector('vc pode comprar armas aqui')),
BadIntentions.ILLEGAL_STUFF)
self.assertEqual(
get_bad_intention(get_text_vector('aqui vende drogas')),
BadIntentions.ILLEGAL_STUFF)
def test_predict_racism_and_xenophobia(self):
self.assertEqual(
get_bad_intention(get_text_vector('sai pra la seu macaco')),
BadIntentions.RACISM_XENOPHOBIA)
self.assertEqual(
get_bad_intention(get_text_vector('seu rpeto fedido')),
BadIntentions.RACISM_XENOPHOBIA)
self.assertEqual(
get_bad_intention(
get_text_vector('vc não devia ser chamado de humano')),
BadIntentions.RACISM_XENOPHOBIA)
self.assertEqual(get_bad_intention(get_text_vector('china burro')),
BadIntentions.RACISM_XENOPHOBIA)
def test_predict_what_am_i(self):
"""
Verify that the classifier predicts correctly texts that refers to
Luci being.
"""
self.assertEqual(
get_myself_intention(get_text_vector('vc é uma criança?')),
MyselfIntentions.WHAT_AM_I)
self.assertEqual(
get_myself_intention(get_text_vector('vc é uma pessoa?')),
MyselfIntentions.WHAT_AM_I)
self.assertEqual(get_myself_intention(get_text_vector('vc é um bot?')),
MyselfIntentions.WHAT_AM_I)
def test_predict_praise(self):
self.assertEqual(
get_good_intention(get_text_vector('Vc gosta de maçã?')),
GoodIntentions.PRAISE
)
self.assertEqual(
get_good_intention(get_text_vector('O que vc gosta de comer?')),
GoodIntentions.PRAISE
)
self.assertEqual(
get_good_intention(get_text_vector('qual sua fruta favorita?')),
GoodIntentions.PRAISE
)
def test_predict_grandma(self):
self.assertEqual(
get_my_parents_intention(get_text_vector('Vc tem avó?')),
AboutMyParents.GRANDMA)
self.assertEqual(
get_my_parents_intention(get_text_vector('Quem é sua avó?')),
AboutMyParents.GRANDMA)
self.assertEqual(
get_my_parents_intention(get_text_vector('como é seu avó?')),
AboutMyParents.GRANDMA)
self.assertEqual(
get_my_parents_intention(
get_text_vector('como é o nome da sua avó?')),
AboutMyParents.GRANDMA)
def test_predict_my_age(self):
"""
Verify that the classifier preedicts correctly text inputs that
refers to Luci's age.
"""
self.assertEqual(
get_myself_intention(get_text_vector('quantos anos você tem?')),
MyselfIntentions.MY_AGE)
self.assertEqual(
get_myself_intention(get_text_vector('qual é a sua idade?')),
MyselfIntentions.MY_AGE)
self.assertEqual(
get_myself_intention(get_text_vector('quantos anos tem mesmo?')),
MyselfIntentions.MY_AGE)
def test_predict_about_my_parents(self):
"""
Verify that the global intention classifier predicts correctly texts
that refers to Luci's parentship.
"""
self.assertEqual(
get_global_intention(get_text_vector('você tem pai?')),
GlobalIntentions.ABOUT_MY_PARENTS)
self.assertEqual(
get_global_intention(get_text_vector('você tem mãe?')),
GlobalIntentions.ABOUT_MY_PARENTS)
self.assertEqual(
get_global_intention(get_text_vector('quem são seus pais')),
GlobalIntentions.ABOUT_MY_PARENTS)
def test_predict_my_dad(self):
self.assertEqual(
get_my_parents_intention(get_text_vector('Vc tem pai?')),
AboutMyParents.MY_DAD)
self.assertEqual(
get_my_parents_intention(get_text_vector('Quem é seu pai?')),
AboutMyParents.MY_DAD)
self.assertEqual(
get_my_parents_intention(get_text_vector('como é seu pai?')),
AboutMyParents.MY_DAD)
self.assertEqual(
get_my_parents_intention(
get_text_vector('como é o nome do seu pai?')),
AboutMyParents.MY_DAD)
def test_predict_how_im_feeling(self):
"""
Verify that the classifier predicts correctly texts that refers to
Luci's feelings.
"""
self.assertEqual(
get_myself_intention(get_text_vector('tudo bem com vc?')),
MyselfIntentions.HOW_IM_FEELING)
self.assertEqual(
get_myself_intention(get_text_vector('como você está?')),
MyselfIntentions.HOW_IM_FEELING)
self.assertEqual(
get_myself_intention(get_text_vector('como está se sentindo?')),
MyselfIntentions.HOW_IM_FEELING)
def test_predict_my_mom(self):
self.assertEqual(
get_my_parents_intention(get_text_vector('Vc tem mãe?')),
AboutMyParents.MY_MOTHER)
self.assertEqual(
get_my_parents_intention(get_text_vector('Quem é sua mae?')),
AboutMyParents.MY_MOTHER)
self.assertEqual(
get_my_parents_intention(get_text_vector('como é sua mae?')),
AboutMyParents.MY_MOTHER)
self.assertEqual(
get_my_parents_intention(
get_text_vector('como é o nome da sua mae?')),
AboutMyParents.MY_MOTHER)
def test_predict_sports_and_playing(self):
self.assertEqual(
get_stuff_i_like_intention(get_text_vector('Vc gosta de bricar?')),
StuffILike.SPORTS_AND_PLAYING)
self.assertEqual(
get_stuff_i_like_intention(
get_text_vector('qual sua brincadeira favorita?')),
StuffILike.SPORTS_AND_PLAYING)
self.assertEqual(
get_stuff_i_like_intention(
get_text_vector('Vc gosta de esportes?')),
StuffILike.SPORTS_AND_PLAYING)
self.assertEqual(
get_stuff_i_like_intention(get_text_vector('gosta de futebol?')),
StuffILike.SPORTS_AND_PLAYING)
def test_predict_about_friends(self):
"""
Verify that the global intention classifier predicts correctly texts
that refers to Luci's freiends.
"""
self.assertEqual(
get_global_intention(get_text_vector('vc tem amigos?')),
GlobalIntentions.ABOUT_MY_FRIENDS)
self.assertEqual(
get_global_intention(get_text_vector('quem é seu melhor amigo?')),
GlobalIntentions.ABOUT_MY_FRIENDS)
self.assertEqual(
get_global_intention(
get_text_vector('nós somos amigos não somos?')),
GlobalIntentions.ABOUT_MY_FRIENDS)
