Add a test

HazyResearch · Oct 22, 2020 · 2c91583 · 2c91583
1 parent 37eead8
commit 2c91583
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/tests/test_figures.py b/tests/test_figures.py
@@ -0,0 +1,11 @@
+"""Test figures."""
+from bs4 import BeautifulSoup
+
+import pdftotree
+
+
+def test_figures():
+    output = pdftotree.parse("tests/input/md.pdf")
+    soup = BeautifulSoup(output, "lxml")
+    imgs = soup.find_all("img")
+    assert len(imgs) == 1