policz_znaki - v2 zagnieżdżenia

dzien6/przyklady_funkcji/policz_znaki.py

 def policz_znaki(napis):
     wynik = 0
-    czy_liczyc = False
+    poziom = 0
     for znak in napis:
         if znak == '<':
-            czy_liczyc = True
+            poziom += 1
         elif znak == '>':
-            czy_liczyc = False
-        elif czy_liczyc:
-            wynik += 1
+            poziom -= 1
+        else:
+            wynik += poziom
     return wynik

dzien6/przyklady_funkcji/policz_znaki_pytest.py

@@ -4,5 +4,8 @@ def test_policz_znaki_jeden_tekst():
     wynik = policz_znaki('ala ma <kota> a kot ma ale')
     assert wynik == 4
 
+def test_policz_znaki_dwa_teksty():
+    assert policz_znaki('Ala ma <kota> a Ola ma <psa>') == 7
+
 def test_policz_znaki_zagniezdzenie():
     assert policz_znaki('a <a<a<aaa>a>>') == 14