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))
