fix bugs in test_cartepizza class

CartePizzeria.py

@@ -1,3 +1,6 @@
+from CartePizzeriaException import CartePizzeriaException
+
+
 class CartePizzeria:
     def __init__(self):
         self.pizzas=[]
@@ -14,7 +17,8 @@ def add_pizza(self,pizza):
     def remove_pizza(self,name):
         for p in self.pizzas:
             if(p.name==name):
-                self.pizzas.remove(p)
+                self.pizzas.remove(p) 
+                return
 
-            raise CartePizzeriaException("Pizza not found: " + name)
+        raise CartePizzeriaException("Pizza not found: " + name)
 

test_cartePizza.py

@@ -0,0 +1,43 @@
+import pytest
+from unittest.mock import MagicMock
+from CartePizzeria import CartePizzeria
+from CartePizzeriaException import CartePizzeriaException
+
+def test_is_empty():
+    carte_pizza=CartePizzeria()
+    assert carte_pizza.is_empty()==True
+
+    pizza_mock=MagicMock()
+    carte_pizza.add_pizza(pizza_mock)
+    assert carte_pizza.is_empty()==False
+
+def test_nb_pizzas():
+    carte_pizza=CartePizzeria()
+    assert carte_pizza.nb_pizzas()==0 
+
+    marguarita=MagicMock()
+    campionne=MagicMock()
+    carte_pizza.add_pizza(marguarita)
+    carte_pizza.add_pizza(campionne)
+    assert carte_pizza.nb_pizzas()==2 
+
+def test_add_pizza():
+    carte_pizza=CartePizzeria()
+    campionne=MagicMock()
+    carte_pizza.add_pizza(campionne)
+    assert carte_pizza.nb_pizzas()==1
+
+def test_remove_pizza():
+    carte_pizza=CartePizzeria()
+    pizza_mock_1=MagicMock()
+    pizza_mock_1.name="vegetarienne"
+    pizza_mock_2=MagicMock()
+    pizza_mock_2.name="marguarita"
+    carte_pizza.add_pizza(pizza_mock_1)  
+    carte_pizza.add_pizza(pizza_mock_2) 
+
+    carte_pizza.remove_pizza("vegetarienne")
+    assert carte_pizza.nb_pizzas()==1
+
+    with pytest.raises(CartePizzeriaException):
+        carte_pizza.remove_pizza("Hawaiiane")