def test_init(self): mess = 'compute table failed' table = CrossTable(self.X, self.Y) crTab = table.getCrosstable() for i in range(self.r): self.assertEqual(all(crTab[i]), all(self.T[i]), mess) self.assertEqual(all(table.getSumRows()), all(self.sum_r), mess) self.assertEqual(all(table.getSumCols()), all(self.sum_s), mess) self.assertEqual(table.n, self.total, mess) r,s = table.shape self.assertEqual(r, self.r, mess) self.assertEqual(s, self.s, mess) x = np.ma.array([1,2,3,4,5,5]) y = np.ma.array([1,2,3,3,5,6]) self.assertRaises(CrossTable(x,y)) table = CrossTable(x,y, expand=True) for i in [1,2,3,4,5,6]: assert i in table.graduation_x assert i in table.graduation_y
def test_init(self): mess = 'compute table failed' table = CrossTable(self.X, self.Y) crTab = table.getCrosstable() for i in range(self.r): self.assertEqual(all(crTab[i]), all(self.T[i]), mess) self.assertEqual(all(table.getSumRows()), all(self.sum_r), mess) self.assertEqual(all(table.getSumCols()), all(self.sum_s), mess) self.assertEqual(table.n, self.total, mess) r, s = table.shape self.assertEqual(r, self.r, mess) self.assertEqual(s, self.s, mess) x = np.ma.array([1, 2, 3, 4, 5, 5]) y = np.ma.array([1, 2, 3, 3, 5, 6]) self.assertRaises(CrossTable(x, y)) table = CrossTable(x, y, expand=True) for i in [1, 2, 3, 4, 5, 6]: assert i in table.graduation_x assert i in table.graduation_y