yujunjun/olmocr
1
0
Fork 0
mirror of https://github.com/allenai/olmocr.git synced 2025-12-17 18:26:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

CI

Browse Source
This commit is contained in:
Jake Poznanski 2025-02-14 20:51:04 +00:00
parent 25ec87b66d
commit 58bdfa512b
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/test_anchor.py
View File

@ -133,7 +133,7 @@ class AnchorTest(unittest.TestCase):
print(anchor_text) print(anchor_text)
print(len(anchor_text)) print(len(anchor_text))
self.assertLess(len(anchor_text), 4000) self.assertLessEqual(len(anchor_text), 4000)
def testEmptyAnchor(self): def testEmptyAnchor(self):
local_pdf_path = os.path.join(os.path.dirname(__file__), "gnarly_pdfs", "tobacco_missed_tokens_pg1.pdf") local_pdf_path = os.path.join(os.path.dirname(__file__), "gnarly_pdfs", "tobacco_missed_tokens_pg1.pdf")
@ -151,7 +151,7 @@ class AnchorTest(unittest.TestCase):
print(anchor_text) print(anchor_text)
print(len(anchor_text)) print(len(anchor_text))
self.assertLess(len(anchor_text), 6000) self.assertLessEqual(len(anchor_text), 6000)
@unittest.skip("TODO, this unit test still fails, the map text is too large.") @unittest.skip("TODO, this unit test still fails, the map text is too large.")
def testExcessiveMapAnchor(self): def testExcessiveMapAnchor(self):
@ -161,7 +161,7 @@ class AnchorTest(unittest.TestCase):
print(anchor_text) print(anchor_text)
print(len(anchor_text)) print(len(anchor_text))
self.assertLess(len(anchor_text), 4000) self.assertLessEqual(len(anchor_text), 4000)
def testKyleOnePageAnchors1(self): def testKyleOnePageAnchors1(self):
local_pdf_path = os.path.join(os.path.dirname(__file__), "gnarly_pdfs", "dolma-page-1.pdf") local_pdf_path = os.path.join(os.path.dirname(__file__), "gnarly_pdfs", "dolma-page-1.pdf")
@ -170,7 +170,7 @@ class AnchorTest(unittest.TestCase):
print(anchor_text) print(anchor_text)
print(len(anchor_text)) print(len(anchor_text))
self.assertLess(len(anchor_text), 6000) self.assertLessEqual(len(anchor_text), 6000)
def testKyleOnePageAnchors2(self): def testKyleOnePageAnchors2(self):
local_pdf_path = os.path.join(os.path.dirname(__file__), "gnarly_pdfs", "olmo-page-1.pdf") local_pdf_path = os.path.join(os.path.dirname(__file__), "gnarly_pdfs", "olmo-page-1.pdf")
@ -179,7 +179,7 @@ class AnchorTest(unittest.TestCase):
print(anchor_text) print(anchor_text)
print(len(anchor_text)) print(len(anchor_text))
self.assertLess(len(anchor_text), 6000) self.assertLessEqual(len(anchor_text), 6000)
class BuildSilverTest(unittest.TestCase): class BuildSilverTest(unittest.TestCase):