Pracownik - zerowanie

dzien7/klasy_wprowadzenie/pracownik1.py

@@ -33,8 +33,10 @@ class Pracownik:
             self.suma_godzin += godziny
 
     def wyplata(self):
-        return self.suma_godzin * self.stawka + self.suma_nadgodzin * self.stawka * 2
-
+        wynik = self.suma_godzin * self.stawka + self.suma_nadgodzin * self.stawka * 2
+        self.suma_godzin = 0
+        self.suma_nadgodzin = 0
+        return wynik
 
 
 def test_jedna_praca():
@@ -43,7 +45,6 @@ def test_jedna_praca():
     wynik = p.wyplata()
     assert wynik == 600
 
-
 def test_suma_prac():
     p = Pracownik('Jan', 'Nowak', 100.0)
     p.praca(6)
@@ -51,7 +52,6 @@ def test_suma_prac():
     wynik = p.wyplata()
     assert wynik == 1000
 
-
 def test_nadgodziny():
     p = Pracownik('Jan', 'Nowak', 100.0)
     p.praca(10)
@@ -59,12 +59,28 @@ def test_nadgodziny():
     wynik = p.wyplata()
     assert wynik == 1700
 
+def test_zero():
+    p = Pracownik('Jan', 'Nowak', 100.0)
+    p.praca(10)
+    p.praca(5)
+    wynik1 = p.wyplata()
+    # w tym teście nie sprawdzamy wartości pierwszej wypłaty - tym zajmują się inne testy
+    wynik2 = p.wyplata()
+    # w tym teście sprawdzamy, czy kolejna wypłata daje zero
+    assert wynik2 == 0
+
+
 def main():
     p = Pracownik('Jan', 'Nowak', 100.0)
     p.praca(6)
     p.praca(4)
     wynik = p.wyplata()
     print(wynik)
+    wynik = p.wyplata()
+    print(wynik)
+    p.praca(10)
+    wynik = p.wyplata()
+    print(wynik)
 
 if __name__ == '__main__':
     main()