From e5d7fe888fef07a79faac05e3df970dfc75a6486 Mon Sep 17 00:00:00 2001
From: Tim Grelka <grelka@indibit.eu>
Date: Wed, 11 Dec 2024 01:09:53 +0100
Subject: [PATCH] refactor: use paragraph test helper

---
 test/test_comment.py |  7 ++++---
 test/test_morse.py   | 16 ++++++++--------
 2 files changed, 12 insertions(+), 11 deletions(-)

diff --git a/test/test_comment.py b/test/test_comment.py
index a090a6e..a9be596 100644
--- a/test/test_comment.py
+++ b/test/test_comment.py
@@ -2,18 +2,19 @@ import mistletoe
 
 from renderer.fifty_ohm_html_renderer import FiftyOhmHtmlRenderer
 from renderer.fifty_ohm_latex_renderer import FiftyOhmLaTeXRenderer
+from test.util import paragraph
 
 
 def test_comment_html():
     assertions = {
-        "%Comment\nBar": "<p>Bar</p>\n",
+        "%Comment\nBar": "Bar",
         # TODO: This test is not working as expected. The comment should be removed
         # "%Comment\n" : "a",
-        "Foo\n%Comment\nBar": "<p>Foo\nBar</p>\n",
+        "Foo\n%Comment\nBar": "Foo\nBar",
     }
 
     for assertion in assertions:
-        assert mistletoe.markdown(assertion, FiftyOhmHtmlRenderer) == assertions[assertion]
+        assert mistletoe.markdown(assertion, FiftyOhmHtmlRenderer) == paragraph(assertions[assertion])
 
 
 def test_comment_latex():
diff --git a/test/test_morse.py b/test/test_morse.py
index d7ece1a..fa9b86a 100644
--- a/test/test_morse.py
+++ b/test/test_morse.py
@@ -1,17 +1,17 @@
 import mistletoe
 
 from renderer.fifty_ohm_html_renderer import FiftyOhmHtmlRenderer
+from test.util import paragraph
 
 
-def test_morse_html() :
-
+def test_morse_html():
     assertions = {
-        '"[morse:a]"' : "<p>"+FiftyOhmHtmlRenderer.render_morse_helper([[1,2]])+"</p>\n",
-        '"[morse: a]"' : "<p>"+FiftyOhmHtmlRenderer.render_morse_helper([[1,2]])+"</p>\n",
-        '"[morse: aaa]"' : "<p>"+FiftyOhmHtmlRenderer.render_morse_helper([[1,2],[1,2],[1,2]])+"</p>\n",
-        '"[morse: a a]"' : "<p>"+FiftyOhmHtmlRenderer.render_morse_helper([[1,2],[3],[1,2]])+"</p>\n",
-        '"[morse: db0]"' : "<p>"+FiftyOhmHtmlRenderer.render_morse_helper([[2,1,1],[2,1,1,1],[2,2,2,2,2]])+"</p>\n",
+        '"[morse:a]"': FiftyOhmHtmlRenderer.render_morse_helper([[1, 2]]),
+        '"[morse: a]"': FiftyOhmHtmlRenderer.render_morse_helper([[1, 2]]),
+        '"[morse: aaa]"': FiftyOhmHtmlRenderer.render_morse_helper([[1, 2], [1, 2], [1, 2]]),
+        '"[morse: a a]"': FiftyOhmHtmlRenderer.render_morse_helper([[1, 2], [3], [1, 2]]),
+        '"[morse: db0]"': FiftyOhmHtmlRenderer.render_morse_helper([[2, 1, 1], [2, 1, 1, 1], [2, 2, 2, 2, 2]]),
     }
 
     for assertion in assertions:
-        assert mistletoe.markdown(assertion, FiftyOhmHtmlRenderer) == assertions[assertion]
\ No newline at end of file
+        assert mistletoe.markdown(assertion, FiftyOhmHtmlRenderer) == paragraph(assertions[assertion])