Add test_markup()

tanghaibao · May 11, 2024 · 385571d · 385571d
1 parent 4d55b17
commit 385571d
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/jcvi/graphics/base.py b/jcvi/graphics/base.py
@@ -413,6 +413,8 @@ def markup(s: str):
     """
     Change the string to latex format, and italicize the text between *.
     """
+    if not rcParams["text.usetex"]:
+        return s
     if "$" in s:
         return s
     s = latex(s)

diff --git a/tests/graphics/test_base.py b/tests/graphics/test_base.py
@@ -3,6 +3,8 @@
 
 import pytest
 
+from jcvi.graphics.base import latex, markup, rc
+
 
 @pytest.mark.parametrize(
     "s,expected",
@@ -24,15 +26,21 @@ def test_shorten(s, expected):
 @pytest.mark.parametrize(
     "s,expected",
     [
-        ("grape_grape vs peach_peach", "grape\_grape vs peach\_peach"),
+        ("grape_grape vs peach_peach", r"grape\_grape vs peach\_peach"),
     ],
 )
 def test_latex(s, expected):
-    from jcvi.graphics.base import latex
-
     assert latex(s) == expected, "Expect {}".format(expected)
 
 
+def test_markup():
+    rc("text", usetex=True)
+    s = "Prupe_1G289800.1"
+    assert markup(s) == r"Prupe\_1G289800.1"
+    rc("text", usetex=False)
+    assert markup(s) == s
+
+
 @pytest.mark.parametrize(
     "figname,format,expected",
     [