def test_crosstab(self): l1= [1, 2, 7, 7, 2, 1, 2, 1, 1] l2= [1, 3, 2, 6, 6, 3, 6, 4, 4] correct = np.array([('1', 1, 0, 1, 2, 0), ('2', 0, 0, 1, 0, 2), ('7', 0, 1, 0, 0, 1)], dtype=[('col1_value', 'S1'), ('1', int), ('2', int), ('3', int), ('4', int), ('6', int)]) self.assertTrue(np.array_equal(correct, crosstab(l1,l2)))
def test_crosstab(self): l1= [1, 2, 7, 7, 2, 1, 2, 1, 1] l2= [1, 3, 2, 6, 6, 3, 6, 4, 4] correct = np.array([('1', 1, 0, 1, 2, 0), ('2', 0, 0, 1, 0, 2), ('7', 0, 1, 0, 0, 1)], dtype=[('col1_value', 'S1'), ('1', int), ('2', int), ('3', int), ('4', int), ('6', int)]) correct_printout = """ col1_value 1 2 3 4 6 0 1 1 0 1 2 0 1 2 0 0 1 0 2 2 7 0 1 0 0 1 """.strip() with uft.rerout_stdout() as get_stdout: self.assertTrue(np.array_equal(correct, crosstab(l1,l2))) self.assertEqual(get_stdout().strip(), correct_printout)