Commitcd295de

authored

Robust pdf loading for empty pages (#115)

* more robust PDF loading for "empty PDFs"* Tests* fix tests

1 parent79bd1af commitcd295deCopy full SHA for cd295de

File tree

+12

-1

lines changed

+12

-1

lines changed

Lines changed: 4 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -53,6 +53,10 @@ def extract_words_for_page(`
`53`	`53`	`)`
`54`	`54`
`55`	`55`	`df=pd.DataFrame(tokens)`
	`56`	`+`
	`57`	`+iflen(df)==0:`
	`58`	`+returnLayout()`
	`59`	`+`
`56`	`60`	`df[["x0","x1"]]= (`
`57`	`61`	`df[["x0","x1"]].clip(lower=0,upper=int(page.width)).astype("float")`
`58`	`62`	`)`

17.8 KB

Binary file not shown.

Lines changed: 8 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -78,4 +78,11 @@ def test_pdf():`
`78`	`78`	`assertattr_nameinpage_layout.page_data`
`79`	`79`
`80`	`80`	`assertlen(set(ele.typeforeleinpage_layout))==3`
`81`		`-# Only three types of font show-up in the file`
	`81`	`+# Only three types of font show-up in the file`
	`82`	`+`
	`83`	`+deftest_empty_pdf():`
	`84`	`+pdf_layout=load_pdf("tests/fixtures/io/empty.pdf")`
	`85`	`+assertlen(pdf_layout)==1# Only one page`
	`86`	`+`
	`87`	`+page_layout=pdf_layout[0]`
	`88`	`+assertlen(page_layout)==0# No selectable tokens on the page`

Comments

(0)