From 2838c8caf0c5ec51eafdc83bfeb4894535741a24 Mon Sep 17 00:00:00 2001 From: Dima Gerasimov Date: Tue, 17 Sep 2019 17:42:21 +0400 Subject: [PATCH] allow passing pdfs for testing --- reading/pdfs.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/reading/pdfs.py b/reading/pdfs.py index 9df7552..56af874 100755 --- a/reading/pdfs.py +++ b/reading/pdfs.py @@ -116,11 +116,12 @@ def test2(): print(res) -def get_annotated_pdfs() -> List[Pdf]: +def get_annotated_pdfs(pdfs=None) -> List[Pdf]: logger = get_logger() setup_logzero(logger, level=logging.DEBUG) - pdfs = get_candidates() + if pdfs is None: + pdfs = get_candidates() logger.info('processing %d pdfs', len(pdfs)) collected = []