test for parse md

telekom · Nov 27, 2023 · f932c6a · f932c6a
1 parent a3cefb8
commit f932c6a
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/tests/test_md.py b/tests/test_md.py
@@ -4,7 +4,7 @@
 
 from typing import Final
 
-from mltb2.md import MdTextSplitter, _chunk_md_by_headline, chunk_md
+from mltb2.md import MdTextSplitter, _chunk_md_by_headline, chunk_md, extract_text_from_markdown
 from mltb2.transformers import TransformersTokenCounter
 
 MD: Final[
@@ -28,6 +28,14 @@
 
 #### Headline 4 / 2"""
 
+def test_():
+    tests = [('## Basic list','Basic list'), 
+            ('* fruit', 'fruit'), 
+            ('> A single quote', 'A single quote'),
+            ('[Link text Here](https://link-url-here.org)','Link text Here')]
+    for md_text, text in tests:
+        assert extract_text_from_markdown(md_text) == text
+
 
 def test_chunk_md_by_headline():
     result = _chunk_md_by_headline(MD)