示例#1
0
    def test_parse_restricoes_turma(self):
        self.assertEqual(len(Hora.instances.values()), 0)
        self.assertEqual(len(Horario.instances.values()), 0)
        self.assertEqual(len(Turma.instances.values()), 0)

        dt = DataParser("data/Escola_A.xlsx")
        dt.parse_dados()
        dt.parse_configuracoes()
        dt.parse_restricoes_turma()

        self.assertEqual(len(Hora.instances.values()), 6)
        self.assertEqual(len(Horario.instances.values()), 30)
        self.assertEqual(len(Turma.instances.values()), 11)

        turma_1 = Turma('1')
        self.assertEqual(len(turma_1.restricoes), 2)
        self.assertIn(Horario.get(Horario.construir_identificador('Terça', Hora('11:30:00'))),
                      turma_1.restricoes)
        self.assertIn(Horario.get(Horario.construir_identificador('Segunda', Hora('11:30:00'))),
                      turma_1.restricoes)

        turma_8 = Turma('8')
        self.assertEqual(len(turma_8.restricoes), 3)
        self.assertIn(Horario.get(Horario.construir_identificador('Quarta', Hora('11:30:00'))),
                      turma_8.restricoes)
        self.assertIn(Horario.get(Horario.construir_identificador('Segunda', Hora('11:30:00'))),
                      turma_8.restricoes)
        self.assertIn(Horario.get(Horario.construir_identificador('Quarta', Hora('10:40:00'))),
                      turma_8.restricoes)