def test_average_columns(self): ds = ((1, 2, 3), (4, 5, 6), (7, 8, 9)) expected = (4, 5, 6) self.assertEquals(expected, hr.reduce_columns(ds, lambda x: sum(x)/len(x)))
def test_sum_columns(self): ds = ((1, 2, 3), (4, 5, 6), (7, 8, 9)) expected = (12, 15, 18) self.assertEquals(expected, hr.reduce_columns(ds, sum))
def test_average_columns_no_data(self): ds = () expected = () self.assertEquals(expected, hr.reduce_columns(ds, lambda x: sum(x)/len(x)))
def test_sum_columns_no_data(self): ds = () expected = () self.assertEquals(expected, hr.reduce_columns(ds, sum))