allow passing pdfs for testing

This commit is contained in:
Dima Gerasimov 2019-09-17 17:42:21 +04:00
parent a016e494c2
commit 2838c8caf0

View file

@ -116,11 +116,12 @@ def test2():
print(res) print(res)
def get_annotated_pdfs() -> List[Pdf]: def get_annotated_pdfs(pdfs=None) -> List[Pdf]:
logger = get_logger() logger = get_logger()
setup_logzero(logger, level=logging.DEBUG) setup_logzero(logger, level=logging.DEBUG)
pdfs = get_candidates() if pdfs is None:
pdfs = get_candidates()
logger.info('processing %d pdfs', len(pdfs)) logger.info('processing %d pdfs', len(pdfs))
collected = [] collected = []