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
