fix remove_indentation bug

CHANGELOG.md

@@ -1,5 +1,9 @@
 # 1.0.6
 
+- Fix: removing indentation bug
+
+# 1.0.6
+
 - Fix: (#26) bug in note empty line stripping, thanks @Jaschenn for reporting
 - New: get_references_for_sql table method
 

pydbml/tools.py

@@ -43,6 +43,9 @@ def strip_empty_lines(source: str) -> str:
 
 
 def remove_indentation(source: str) -> str:
+    if not source:
+        return source
+
     pattern = re.compile(r'^\s*')
 
     lines = source.split('\n')

test/test_tools.py

@@ -1,7 +1,7 @@
 from unittest import TestCase
 
 from pydbml.classes import Note
-from pydbml.tools import comment_to_dbml
+from pydbml.tools import comment_to_dbml, remove_indentation
 from pydbml.tools import comment_to_sql
 from pydbml.tools import indent
 from pydbml.tools import note_option_to_dbml
@@ -98,4 +98,15 @@ def test_one_empty_line(self) -> None:
     def test_end(self) -> None:
         stripped = '   line1\n\n line2'
         source = f'\n{stripped}\n   '
-        self.assertEqual(strip_empty_lines(source), stripped)
+        self.assertEqual(strip_empty_lines(source), stripped)
+
+
+class TestRemoveIndentation(TestCase):
+    def test_empty(self) -> None:
+        source = ''
+        self.assertEqual(remove_indentation(source), source)
+
+    def test_not_empty(self) -> None:
+        source = '    line1\n     line2'
+        expected = 'line1\n line2'
+        self.assertEqual(remove_indentation(source), expected)