#!/usr/bin/env python

"""
illustration of numpy broadcasting
"""

import os
import unittest
import numpy as np

           #apples, beef, eggs, potatos
calories = [[56., 0.0, 4.4, 68.], # carbs
            [1.2, 104., 52., 8.], # protein
            [1.8, 135., 99., 0.9]] # fat

class TestBroadcasting(unittest.TestCase):
    def test_basic(self):
        """example of broadcasting"""

        A = np.array(calories)
        cal = A.sum(axis=0)
        print (cal)
        percentage = 100*A/cal.reshape(1,4)
        print (percentage)

if __name__ == '__main__':
    unittest.main()