Add a test for validate_brackets

dfint · Oct 1, 2023 · 33ea7d5 · 33ea7d5
1 parent f66b7d5
commit 33ea7d5
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/df_translation_toolkit/validation/validate_objects.py b/df_translation_toolkit/validation/validate_objects.py
@@ -1,7 +1,7 @@
 from df_translation_toolkit.parse.parse_raws import all_caps, split_tag
 
 
-def validate_brackets(tag: str):
+def validate_brackets(tag: str) -> bool:
     return tag.startswith("[") and tag.endswith("]") and tag.count("[") == 1 and tag.count("]") == 1
 
 

diff --git a/tests/test_validate_objects.py b/tests/test_validate_objects.py
@@ -0,0 +1,16 @@
+import pytest
+
+from df_translation_toolkit.validation.validate_objects import validate_brackets
+
+
+@pytest.mark.parametrize(
+    "text, expected",
+    [
+        ("[text]", True),
+        ("[text", False),
+        ("text]", False),
+        ("[text[]", False),
+    ]
+)
+def test_validate_brackets(text, expected):
+    assert validate_brackets(text) == expected