def testa_prevalencias_somam_um(self): prev, tot = calcula_prevalencias(carrega_dados()) self.assertAlmostEqual(prev.processo.sum(), 1, 3) self.assertAlmostEqual(prev.corte.sum(), 1, 3) self.assertAlmostEqual(prev.toi.sum(), 1, 3) self.assertAlmostEqual(prev.negativacao.sum(), 1, 3) self.assertAlmostEqual(prev.reclamacao.sum(), 1, 3) self.assertAlmostEqual(prev.toicorte.sum(), 1, 3) self.assertAlmostEqual(prev.toineg.sum(), 1, 3)
def testa_tamanho_tabela_prevalência(self): prev, tot = calcula_prevalencias(carrega_dados()) self.assertEqual(len(prev), 31) self.assertEqual(len(tot), 31)