BetterTestCov: Tests for shape

Signed-off-by: Gabe Goodhart <[email protected]>

Author: gabe-l-hart

scriptit/shape.py

@@ -49,9 +49,11 @@ def progress_bar(
     Returns:
         progress_bar_str (str): String for the progress bar
     """
-    assert len(done_char) == 1
-    assert len(undone_char) == 1
-    assert len(head_char) == 1
+    if len(done_char) != 1 or len(undone_char) != 1 or len(head_char) != 1:
+        raise ValueError("All char args must have length 1")
+    # Flatten to [0, 1]. This is done instead of raising since this may be
+    # computed and should not break
+    complete_pct = max(min(1.0, complete_pct), 0.0)
     if width is None:
         width = shutil.get_terminal_size().columns
     n_done = int((width - 3) * complete_pct)

tests/test_shape.py

@@ -0,0 +1,31 @@
+"""
+Tests for the shape module
+"""
+# Standard
+import re
+
+# Third Party
+import pytest
+
+# Local
+from scriptit import shape
+
+
+def test_progress_bar():
+    """Test various progress_bar configurations"""
+    assert shape.progress_bar(0.5, 10) == "[===>----]"
+    assert (
+        shape.progress_bar(0.5, 10, done_char=">", undone_char=".", head_char="|")
+        == "[>>>|....]"
+    )
+    assert shape.progress_bar(-1, 10) == "[>-------]"
+    assert shape.progress_bar(2, 10) == "[=======>]"
+    match = re.match(r"\[(=+)>(-)+\]", shape.progress_bar(0.75))
+    assert match
+    assert len(match.group(1)) > len(match.group(2))
+    with pytest.raises(ValueError):
+        shape.progress_bar(0.5, 10, done_char="xx")
+
+
+def test_box():
+    """Test various"""