Merge pull request #84 from Geson-anko/name_tools/count_head_sep

Name tools/count head sep

JarvisEngine/core/name.py

@@ -13,7 +13,21 @@ def clean(name:str) -> str:
         end = len(name)
 
     return name[start:end]
-
+
+def count_head_sep(name:str) -> int:
+    """count number of head separator of name.
+    Ex:
+        "..a.b" -> 2
+        "...c" -> 3
+    """
+    num = 0
+    for s in name:
+        if s != SEP:
+            return num
+        else:
+            num += 1
+    return num
+
 def join(name:str, *others:str) -> str:
     """join names
     Ex: join("a.b.","c.d") -> "a.b.c.d"

tests/core/test_name.py

@@ -14,4 +14,12 @@ def test_join():
     assert name.join("a.b.c") == "a.b.c"
     assert name.join(".a.b.","c.",".d") == "a.b.c.d"
     assert name.join("a","b","c","d") == "a.b.c.d"
-    assert name.join(".a.b.c.d") == "a.b.c.d"
+    assert name.join(".a.b.c.d") == "a.b.c.d"
+
+def test_count_head_sep():
+    assert name.count_head_sep("..a.b.") == 2
+    assert name.count_head_sep("") == 0
+    assert name.count_head_sep(".....") == 5
+    assert name.count_head_sep("asf....") == 0
+    assert name.count_head_sep("..v..s..a.sd") == 2
+