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))