unittest¶

class CompteNonApprovisionne(Exception):
    """Pas assez d'argent sur le compte..."""

class Compte:
    """Un compte bancaire"""
    def __init__(self):
    self.solde = 0
    def credite(self, amount):
    self.solde += amount
    def debite(self, amount):
    if self.solde < amount:
        raise CompteNonApprovisionne()
    self.solde -= amount

from compte import *
import unittest

class TestCompte(unittest.TestCase):
    def testDebit(self):
    compte = Compte()
    compte.credite(100)
    self.assertRaises(CompteNonApprovisionne, compte.debite, 200)
    self.assertEqual(compte.solde, 100)

    def testCreationCompte(self):
    compte = Compte()
    self.assertEqual(compte.solde, 0)

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