Fixes #67

Semprini · Mar 17, 2024 · ee4b016 · ee4b016
1 parent d858bc5
commit ee4b016
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/mdg/tests/test_tools.py b/mdg/tests/test_tools.py
@@ -12,6 +12,9 @@ def setUp(self):
     def test_camel(self):
         self.assertEqual("test", camelcase("test"))
         self.assertEqual("test", camelcase("Test"))
+        self.assertEqual("test", camelcase("Test "))
+        self.assertEqual("test", camelcase("Test_"))
+        self.assertEqual("test", camelcase(" Test_"))
         self.assertEqual("testCase", camelcase("TestCase"))
         self.assertEqual("testCase", camelcase("testCase"))
         self.assertEqual("testCase", camelcase("test case"))

diff --git a/mdg/tools/case.py b/mdg/tools/case.py
@@ -10,7 +10,7 @@
 def camelcase(string:str) -> str:
     if string is None or string == "":
         return string
-    words: list = re.split(' |_', string)
+    words: list = re.split(' |_', string.strip(' _'))
 
     # Can't use Title function in str library as this lowers all but the first char: does not handle existing camelCased input
     words_camel: list = [(words[0][0].lower() + words[0][1:])] + list(word[0].upper() + word[1:] for word in words[1:])