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)