bite 115

Author: boraxpr

115/indents.py

@@ -0,0 +1,12 @@
+def count_indents(text):
+    """Takes a string and counts leading white spaces, return int count"""
+    counts = 0
+    for char in text:
+        if char.isspace() and char != "\t" and char !="\n":
+            counts += 1
+        elif char.isalpha():
+            break
+    return counts
+
+# print(count_indents("\t\toxzcxc"))
+# "ddd dd dd".

115/test_indents.py

@@ -0,0 +1,16 @@
+import pytest
+
+from indents import count_indents
+
+
+@pytest.mark.parametrize("input_string, count", [
+   ('string  ', 0),
+   ('  string', 2),
+   ('    string', 4),
+   ('            string', 12),
+   ('\t\tstring', 0),
+   ('  str  ing', 2),
+   ('  str  ', 2),
+])
+def test_count_indents(input_string, count):
+    assert count_indents(input_string) == count