TestAverageStack.py

import unittest
from AverageStack import AverageStack

class TestAverageStack(unittest.TestCase):

    def test1(self):
        a = AverageStack(10)
        self.assertEquals(a.avg(), 0)

    def test2(self):
        a = AverageStack(10,10)
        self.assertEquals(a.avg(), 10)

    def test3(self):
        a = AverageStack(10,0)
        self.assertEquals(a.avg(100), 10)

    def test4(self):
        a = AverageStack(11,0)
        for i in range(61):
            a.avg(i)
        self.assertEquals(a.avg(), 55)

    def test5(self):
        a = AverageStack(10,0)
        for i in range(61):
            a.avg(i)
        self.assertEquals(a.avg(), 55.5)


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

# eof