Vik Paruchuri
commited on
Commit
·
e3ecc68
1
Parent(s):
148a038
Fix asserts
Browse files
.github/workflows/ci.yml
CHANGED
|
@@ -10,7 +10,7 @@ jobs:
|
|
| 10 |
- name: Install apt requirements
|
| 11 |
run: |
|
| 12 |
sudo apt-get update
|
| 13 |
-
sudo apt-get install -y libpango-1.0-0
|
| 14 |
- name: Set up Python 3.11
|
| 15 |
uses: actions/setup-python@v4
|
| 16 |
with:
|
|
|
|
| 10 |
- name: Install apt requirements
|
| 11 |
run: |
|
| 12 |
sudo apt-get update
|
| 13 |
+
sudo apt-get install -y libpango-1.0-0 libharfbuzz0b libpangoft2-1.0-0 libgdk-pixbuf2.0-0 libcairo2 libffi-dev shared-mime-info
|
| 14 |
- name: Set up Python 3.11
|
| 15 |
uses: actions/setup-python@v4
|
| 16 |
with:
|
tests/converters/test_ocr_converter.py
CHANGED
|
@@ -35,7 +35,7 @@ def check_bboxes(page: OCRJSONPageOutput, lines):
|
|
| 35 |
|
| 36 |
@pytest.mark.config({"page_range": [0]})
|
| 37 |
def test_ocr_converter(config, model_dict, temp_doc):
|
| 38 |
-
_ocr_converter(config, model_dict, temp_doc,
|
| 39 |
|
| 40 |
|
| 41 |
@pytest.mark.filename("pres.pdf")
|
|
|
|
| 35 |
|
| 36 |
@pytest.mark.config({"page_range": [0]})
|
| 37 |
def test_ocr_converter(config, model_dict, temp_doc):
|
| 38 |
+
_ocr_converter(config, model_dict, temp_doc, 83, 2)
|
| 39 |
|
| 40 |
|
| 41 |
@pytest.mark.filename("pres.pdf")
|
tests/providers/test_pdf_provider.py
CHANGED
|
@@ -8,7 +8,7 @@ def test_pdf_provider(doc_provider):
|
|
| 8 |
assert doc_provider.get_images([0], 96)[0].size == (816, 1056)
|
| 9 |
|
| 10 |
page_lines = doc_provider.get_page_lines(0)
|
| 11 |
-
assert len(page_lines) ==
|
| 12 |
|
| 13 |
spans = page_lines[0].spans
|
| 14 |
assert len(spans) == 2
|
|
|
|
| 8 |
assert doc_provider.get_images([0], 96)[0].size == (816, 1056)
|
| 9 |
|
| 10 |
page_lines = doc_provider.get_page_lines(0)
|
| 11 |
+
assert len(page_lines) == 85
|
| 12 |
|
| 13 |
spans = page_lines[0].spans
|
| 14 |
assert len(spans) == 2
|