14.05.09

reload(GraphClass)
import unittest
import GraphClass

class TestGraphClass(unittest.TestCase):

   

   

 

    goodValues= ( ("s1",0),
                  ("s2",1),
                  ("s3",1),
                   ("s4",2),
                   ("s5",2),
                )

    def setUp(self):
        gDict= { "s1" : { "s2" : "s1-s2" ,  "s5" : "s1-s5" },
      "s2" :  { "s3" : "s2-s3" , "s4" : "s2-s4" , "s5" : "s2-s5"},
      "s5" : {"s4" : "s5-s4" }}
        self.G = GraphClass.GraphClass("essai",gDict)

    def testForSuccess(self):
        for s , expectedResult in self.goodValues:
            computedResult=self.G.calculeNbAntecedents(s)
            self.assertEqual(expectedResult,computedResult)

    def testRaisesBadVertexNameException(self):
        self.assertRaises( GraphClass.BadVertexNameException,self.G.calculeNbAntecedents,"NomQuiCommencePasparUnS")

    def testUnknownVertexException(self):
        self.assertRaises( GraphClass.UnknownVertexException,self.G.calculeNbAntecedents,"s10000")

if __name__ == '__main__':
##    unittest.main()
    suite=unittest.TestLoader().loadTestsFromTestCase(TestGraphClass)
    unittest.TextTestRunner(verbosity=2).run(suite)