policz_znaki - v3 - inne znaki

dzien6/przyklady_funkcji/policz_znaki.py

-def policz_znaki(napis):
+def policz_znaki(napis, lewy='<', prawy='>'):
     wynik = 0
     poziom = 0
     for znak in napis:
-        if znak == '<':
+        if znak == lewy:
             poziom += 1
-        elif znak == '>':
+        elif znak == prawy:
             poziom -= 1
         else:
             wynik += poziom

dzien6/przyklady_funkcji/policz_znaki_print.py

@@ -3,3 +3,4 @@ from policz_znaki import *
 print(policz_znaki('ala ma <kota> a kot ma ale'))
 print(policz_znaki('a <a<a<a>>>'))
 print(policz_znaki('a <a<a<aaa>a>>'))
+print(policz_znaki('ala [kota [a kot]] ma [ale]', '[', ']'))
\ No newline at end of file

dzien6/przyklady_funkcji/policz_znaki_pytest.py

@@ -9,3 +9,6 @@ def test_policz_znaki_dwa_teksty():
 
 def test_policz_znaki_zagniezdzenie():
     assert policz_znaki('a <a<a<aaa>a>>') == 14
+
+def test_policz_znaki_inne():
+    assert policz_znaki('ala [kota [a kot]] ma [ale]', '[', ']') == 18