def process_data(matrix): transformed_data = TransformedData() transform_matrix(matrix) transformed_data.matrix_months = matrix transformed_data.total_year = maxarray.sum_total_columns(matrix, range(1,12)) transformed_data.quarter1_total = maxarray.sum_total_columns_from_rows(matrix, range(1,12), axis_rows=range(0,3)) transformed_data.quarter2_total = maxarray.sum_total_columns_from_rows(matrix, range(1,12), axis_rows=range(3,6)) transformed_data.quarter3_total = maxarray.sum_total_columns_from_rows(matrix, range(1,12), axis_rows=range(6,9)) transformed_data.quarter4_total = maxarray.sum_total_columns_from_rows(matrix, range(1,12), axis_rows=range(9,12)) return transformed_data
def test_sum_total_columns_from_rows(self): data = [['Jan',4,4,4],['Feb',2,-2,4], ['March', 10,5,10,5], ['April', 4,4,4]] result = maxarray.sum_total_columns_from_rows(data, axis_columns=range(1,4), axis_rows=[0,3]) self.assertEquals([8, 8, 8], result) print result