python unittest

import unittest

from struct import *

class PingAlgorythmTestCase(unittest.TestCase):
    def setUp(self):
        self.netw = Network('testnw' )
        self.comp1 = Computer('1' )
        self.comp2 = Computer('2' )
        self.comp3 = Computer('3' )
        self.comp4 = Computer('4' )
        self.comp5 = Computer('5' )

        self.conn1 = Connection(self.comp1,self.comp2)
        self.conn2 = Connection(self.comp1,self.comp3)
        self.conn3 = Connection(self.comp3,self.comp4)
        self.conn4 = Connection(self.comp4,self.comp5)
        self.conn5 = Connection(self.comp1,self.comp4)

        self.netw.addConnection(self.conn1)
        self.netw.addConnection(self.conn2)
        self.netw.addConnection(self.conn3)
        self.netw.addConnection(self.conn4)
        self.netw.addConnection(self.conn5)

    def testping1(self):
        alg = PingAlgorithm(self.netw, self.comp1, self.comp2)
        self.assertEqual(alg.doAction().time,1)

    def testping2(self):
        alg = PingAlgorithm(self.netw, self.comp2, self.comp5)
        self.assertEquals(alg.doAction().time,3)

    def testping3(self):
        alg = PingAlgorithm(self.netw, self.comp2, self.comp5)
        self.assertEquals(alg.doAction().time,3)

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