class GeneroTest(TestCase):
def setUp(self):
self.parlamentar = Parlamentar(nome='Ana')
self.proposicao = Proposicao(autor_principal='Ana', indexacao='mulher')
self.id_casa_legislativa = 1
self.genero = Genero()
def test_definir_palavra(self):
# temas_frequencia_mulher = self.genero.definir_palavras('F')
self.assertEqual('Ana', self.parlamentar.nome)
self.assertEqual('Ana', self.proposicao.autor_principal)
self.assertEqual(('mulher'), self.proposicao.indexacao)
def test_organiza_lista_palavras(self):
self.dicionario_palavras = {
"Saude": 2,
"Educacao": 1,
"Transporte": 26
}
self.assertEqual(
self.genero.organiza_lista_palavras(self.dicionario_palavras),
[('Transporte', 26), ('Saude', 2), ('Educacao', 1)])
def test_define_chaves_dicionario(self):
self.dicionario_ordenado = ["Saude", "Saude", "Educacao"]
self.assertEqual(
self.genero.define_chaves_dicionario(self.dicionario_ordenado),
[("Saude", 2), ("Educacao", 1)])
def test_agrupa_palavras(self):
self.assertEqual(
self.genero.agrupa_palavras(self.genero, self.id_casa_legislativa),
[])
def test_get_casas_legislativas_com_genero(self):
self.assertEqual(self.genero.get_casas_legislativas_com_genero(), [])
