Add a python test for EscapeLeadingWhitespace.

Author: nedtwigg

python/selfie-lib/selfie_lib/__init__.py

@@ -6,6 +6,7 @@
 from .CacheSelfie import cache_selfie_binary as cache_selfie_binary
 from .CacheSelfie import cache_selfie_json as cache_selfie_json
 from .CommentTracker import CommentTracker as CommentTracker
+from .EscapeLeadingWhitespace import EscapeLeadingWhitespace as EscapeLeadingWhitespace
 from .FS import FS as FS
 from .Lens import Camera as Camera
 from .Lens import CompoundLens as CompoundLens

python/selfie-lib/tests/EscapeLeadingWhitespace_test.py

@@ -0,0 +1,29 @@
+from selfie_lib import EscapeLeadingWhitespace
+
+def test_detection():
+    # not enough to detect
+    assert EscapeLeadingWhitespace.appropriate_for("") == EscapeLeadingWhitespace.ALWAYS
+    assert EscapeLeadingWhitespace.appropriate_for("abc") == EscapeLeadingWhitespace.ALWAYS
+    assert EscapeLeadingWhitespace.appropriate_for("abc\nabc") == EscapeLeadingWhitespace.ALWAYS
+
+    # all spaces -> only tabs need escape
+    assert EscapeLeadingWhitespace.appropriate_for(" ") == EscapeLeadingWhitespace.ONLY_ON_TAB
+    assert EscapeLeadingWhitespace.appropriate_for("  ") == EscapeLeadingWhitespace.ONLY_ON_TAB
+    assert EscapeLeadingWhitespace.appropriate_for("  \n  ") == EscapeLeadingWhitespace.ONLY_ON_TAB
+
+    # all tabs -> only space needs escape
+    assert EscapeLeadingWhitespace.appropriate_for("\t") == EscapeLeadingWhitespace.ONLY_ON_SPACE
+    assert EscapeLeadingWhitespace.appropriate_for("\t\t") == EscapeLeadingWhitespace.ONLY_ON_SPACE
+    assert EscapeLeadingWhitespace.appropriate_for("\t\n\t") == EscapeLeadingWhitespace.ONLY_ON_SPACE
+
+    # it's a mess -> everything needs escape
+    assert EscapeLeadingWhitespace.appropriate_for("\t\n  ") == EscapeLeadingWhitespace.ALWAYS
+
+    # single spaces and tabs -> only tabs need escape
+    test_string = """/*
+ * Copyright
+ */
+interface Foo {
+\tfun bar()
+}"""
+    assert EscapeLeadingWhitespace.appropriate_for(test_string) == EscapeLeadingWhitespace.ALWAYS