Improve unit string in console output

This fixes #24.
resultwizard · Mar 18, 2024 · 59ffc71 · 59ffc71
1 parent dfb865b
commit 59ffc71
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/api/console_stringifier.py b/src/api/console_stringifier.py
@@ -36,8 +36,10 @@ def _modify_unit(self, unit: str) -> str:
         unit = (
             unit.replace(r"\squared", "^2")
             .replace(r"\cubed", "^3")
+            .replace("\\per\\", "/")
             .replace(r"\per", "/")
-            .replace("\\", "")
+            .replace("\\", " ")
+            .strip()
         )
 
         if unit[0] == "/":

diff --git a/tests/playground.py b/tests/playground.py
@@ -32,7 +32,7 @@
 # wiz.res("", 42.0).print()
 # -> Error: "name must not be empty"
 
-wiz.res("a911", 1.0, r"\mm")
+wiz.res("a911", 1.0, r"\mm\s\per\N\kg")
 # a: 1.0 \mm
 
 wiz.res("1 b", 1.0, 0.01, r"\per\mm\cubed").print()