add indent function

Author: MrMaydo

src/java_model.py

@@ -36,3 +36,21 @@ class Field:
     name: str
     type: str
     description: Optional[str] = None
+
+
+@dataclass
+class JavaClass:
+    name: str
+    fields: List[Field]
+    description: Optional[str] = None
+
+
+def indent(level: int, size: int = 4) -> str:
+    if level < 0 or size < 0:
+        raise ValueError("Indent size cannot be lower than 0")
+
+    return " " * size * level
+
+
+
+

tests/test_java_model.py

@@ -0,0 +1,33 @@
+import pytest
+
+from src.java_model import indent
+
+
+def test_indent_zero():
+    assert indent(0) == ""
+
+
+def test_indent_default_size():
+    assert indent(1) == " " * 4
+
+
+def test_indent_multiple_levels():
+    assert indent(3) == " " * 12
+
+
+def test_indent_custom_size():
+    assert indent(2, size=2) == " " * 4
+
+
+def test_indent_size_zero():
+    assert indent(5, size=0) == ""
+
+
+def test_indent_negative_lvl():
+    with pytest.raises(ValueError):
+        indent(level=-1, size=4)
+
+
+def test_indent_negative_size():
+    with pytest.raises(ValueError):
+        indent(level=1, size=-1)