feat: improve tests for unit rendering

test/test_unit.py

@@ -0,0 +1,45 @@
+import mistletoe
+import pytest
+
+from renderer.fifty_ohm_html_renderer import FiftyOhmHtmlRenderer
+
+from .util import paragraph
+
+
+@pytest.mark.html
+def test_basic_units_html():
+    """Test basic units: Insert non-breaking, smaller space."""
+    assertions = {
+        "1 V": "1 V",
+        "0,12 V": "0,12 V",
+        "0,75 kW": "0,75 kW",
+        "33 A": "33 A",
+        "1 Ah": "1 Ah",  # Ah can be a tricky case if the RegEx is wrong, as A might match before Ah, thus rendering the output with an "A" unit
+    }
+
+    for input, output in assertions.items():
+        assert mistletoe.markdown(input, FiftyOhmHtmlRenderer) == paragraph(output)
+
+
+@pytest.mark.html
+def test_translated_units_html():
+    """Test translated units: Replace Ohm with Ω."""
+    assertions = {
+        "50 Ohm": "50 Ω",
+        "100 mOhm": "100 mΩ",
+    }
+
+    for input, output in assertions.items():
+        assert mistletoe.markdown(input, FiftyOhmHtmlRenderer) == paragraph(output)
+
+
+@pytest.mark.html
+def test_special_units_html():
+    """Test special units: Don't put space before ° and %."""
+    assertions = {
+        "100%": "100%",
+        "42,24°": "42,24°",
+    }
+
+    for input, output in assertions.items():
+        assert mistletoe.markdown(input, FiftyOhmHtmlRenderer) == paragraph(output)

test/test_unit_token.py

@@ -1,18 +0,0 @@
-import mistletoe
-
-from renderer.fifty_ohm_html_renderer import FiftyOhmHtmlRenderer
-
-from .util import paragraph
-
-
-def test_unit_html():
-    assertions = {
-        "0,1 V": "0,1 V",
-        "33 A": "33 A",
-        "1 Ah": "1 Ah",
-        "100 mOhm": "100 mΩ",
-        "25 Ohm": "25 Ω",
-    }
-
-    for input, output in assertions.items():
-        assert mistletoe.markdown(input, FiftyOhmHtmlRenderer) == paragraph(output)

test/util.py

@@ -1,2 +1,2 @@
-def paragraph(text) :
+def paragraph(text):
-    return f'<p>{text}</p>\n'
+    return f"<p>{text}</p>\n"