Merge pull request #4 from explosion/add-tests

Added extra tests

prodigy_pdf/__init__.py

@@ -55,7 +55,7 @@ def pdf_image_manual(
 ) -> ControllerComponentsDict:
     """Turns pdfs into images in order to annotate them."""
     # Read in stream as a list for progress bar.
-    if not pdf_folder.exists():
+    if not Path(pdf_folder).exists():
         msg.fail(f"Folder `{pdf_folder}` does not exist.", exits=True)
     pdf_paths = list(Path(pdf_folder).glob("*.pdf"))
     if len(pdf_paths) == 0:

tests/test_basics.py

@@ -1,8 +1,8 @@
 from pathlib import Path
-from prodigy_pdf import generate_pdf_pages, fold_ocr_dashes
+from prodigy_pdf import generate_pdf_pages, fold_ocr_dashes, pdf_image_manual
 
 
-def test_smoke_internal():
+def test_generate_pdf_pages():
     # We know this one PDF has six pages.
     paths = Path("tests/pdfs").glob("*.pdf")
     pages = list(generate_pdf_pages(paths))
@@ -29,3 +29,8 @@ def test_fold_dashes():
 
     expected = "Real-Time Strategy (RTS) games have become an increasingly popular test-bed for modern artificial intelligence techniques. With this rise in popularity has come the creation of several annual competitions, in which AI agents (bots) play the full game of StarCraft: Broodwar by Blizzard Entertainment. The three major annual StarCraft AI Competitions are the Student StarCraft AI Tournament (SSCAIT), the Computational Intelligence in Games (CIG) competition, and the Artificial Intelligence and Interactive Digital Entertainment (AIIDE) competition. In this paper we will give an overview of the current state of these competitions, and the bots that compete in them."
     assert fold_ocr_dashes(going_in) == expected
+
+
+def test_pdf_image_manual():
+    components = pdf_image_manual("xxx", "tests/pdfs", "foo,bar")
+    assert isinstance(next(components["stream"]), dict)