def _inicializa_votacoes(self):
"""Pega votações deste período no banco de dados, filtra por palavras
chave e seta a lista self.votacoes"""
filtro_votacao = filtro.FiltroVotacao(
self.casa_legislativa, self.periodo, self.palavras_chave)
self.votacoes = filtro_votacao.filtra_votacoes()
return self.votacoes
def test_filtra_proposicoes_sem_votacoes(self):
proposicoes = models.Proposicao.objects.filter(casa_legislativa_id=1)
votacoes = models.Votacao.objects.all()
filtro_votacao = filtro.FiltroVotacao()
proposicoes_com_votacoes = filtro_votacao._filtra_proposicoes_com_votacoes(
proposicoes, votacoes)
self.assertEquals(8, len(proposicoes_com_votacoes))
def test_recupera_proposicoes(self):
casa_legislativa = models.CasaLegislativa()
casa_legislativa.id = 1
filtro_votacao = filtro.FiltroVotacao()
proposicoes = filtro_votacao._recupera_proposicoes(casa_legislativa)
self.assertEquals(9, len(proposicoes))
def test_filtra_votacoes_por_palavras_chave(self):
proposicoes = models.Proposicao.objects.filter(casa_legislativa_id=1)
votacoes = models.Votacao.objects.all()
lista_palavras_chave = ['militar', 'guerra', 'escolas', 'pensão']
filtro_votacao = filtro.FiltroVotacao()
votacoes_com_palavras_chave = filtro_votacao._filtra_votacoes_por_palavras_chave(
proposicoes, votacoes, lista_palavras_chave)
self.assertEquals(4, len(votacoes_com_palavras_chave))
def test_verifica_palavras_chave_nao_relacionadas_em_proposicao(self):
proposicao = models.Proposicao.objects.get(id=8)
votacoes = models.Votacao.objects.filter(proposicao_id=8)
lista_palavras_chave = ['cotas', 'educação', 'violência']
filtro_votacao = filtro.FiltroVotacao()
filtra_proposicoes_por_palavras_chave = filtro_votacao._verifica_palavras_chave_em_proposicao(
proposicao, votacoes, lista_palavras_chave)
self.assertFalse(filtra_proposicoes_por_palavras_chave)
def test_palavra_existe_em_descricao_votacao(self):
proposicao = models.Proposicao.objects.get(id=8)
votacoes = models.Votacao.objects.filter(proposicao_id=8)
palavra_chave = 'inglaterra'
filtro_votacao = filtro.FiltroVotacao()
palavra_existe_em_proposicao = filtro_votacao._palavra_existe_em_proposicao(
proposicao, votacoes, palavra_chave)
self.assertTrue(palavra_existe_em_proposicao)
def test_palavra_nao_existe_em_proposicao(self):
proposicao = models.Proposicao.objects.get(id=1)
votacoes = models.Votacao.objects.filter(proposicao_id=1)
palavra_chave = 'corrupcao'
filtro_votacao = filtro.FiltroVotacao()
palavra_existe_em_proposicao = filtro_votacao._palavra_existe_em_proposicao(
proposicao, votacoes, palavra_chave)
self.assertFalse(palavra_existe_em_proposicao)
def test_recupera_votacoes_da_proposicao(self):
proposicao = models.Proposicao()
proposicao.id = 1
votacoes = models.Votacao.objects.all()
filtro_votacao = filtro.FiltroVotacao()
votacoes_da_proposicao = filtro_votacao._recupera_votacoes_da_proposicao(
proposicao, votacoes)
self.assertEquals(1, len(votacoes_da_proposicao))
def _inicializa_votacoes(self):
"""Grab this voting period in the database, filtering by keyword and
arrow self.votacoes list"""
filtro_votacao = filtro.FiltroVotacao(self.casa_legislativa,
self.periodo,
self.palavras_chave)
self.votacoes = filtro_votacao.filtra_votacoes()
return self.votacoes
def test_filtra_votacoes_com_varias_palavras_chave(self):
casa_legislativa = models.CasaLegislativa.objects.get(id=1)
periodo_casa_legislativa = models.PeriodoCasaLegislativa(
date(1989, 02, 02), date(1989, 10, 10))
lista_palavras_chave = ['militar', 'guerra', 'escolas', 'pensão']
filtro_votacao = filtro.FiltroVotacao()
votacoes_filtradas = filtro_votacao.filtra_votacoes(
casa_legislativa, periodo_casa_legislativa, lista_palavras_chave)
self.assertEquals(4, len(votacoes_filtradas))
def test_filtra_votacoes_sem_palavras_chave_relacionadas(self):
casa_legislativa = models.CasaLegislativa.objects.get(id=1)
periodo_casa_legislativa = models.PeriodoCasaLegislativa(
date(1989, 02, 02), date(1989, 10, 10))
lista_palavras_chave = ['cotas', 'educacao', 'violência']
filtro_votacao = filtro.FiltroVotacao()
votacoes_filtradas = filtro_votacao.filtra_votacoes(
casa_legislativa, periodo_casa_legislativa, lista_palavras_chave)
self.assertEquals(0, len(votacoes_filtradas))
def test_filtra_votacoes_com_periodo_(self):
casa_legislativa = models.CasaLegislativa.objects.get(id=1)
periodo_casa_legislativa = models.PeriodoCasaLegislativa(
date(1989, 8, 8), date(1992, 11, 11))
lista_palavras_chave = ['cotas', 'guerra', 'violência']
filtro_votacao = filtro.FiltroVotacao()
votacoes_filtradas = filtro_votacao.filtra_votacoes(
casa_legislativa, periodo_casa_legislativa, lista_palavras_chave)
self.assertEquals(1, len(votacoes_filtradas))
