Skip to content
This repository was archived by the owner on Mar 6, 2026. It is now read-only.

Commit 678b2d1

Browse files
committed
chore(tests): remove unused test samples and added more tests in test_page
1 parent e9da20b commit 678b2d1

File tree

3 files changed

+33
-72
lines changed

3 files changed

+33
-72
lines changed

samples/snippets/test_token_confidence_sample.py

Lines changed: 0 additions & 36 deletions
This file was deleted.

samples/snippets/test_token_detected_languages_sample.py

Lines changed: 0 additions & 36 deletions
This file was deleted.

tests/unit/test_page.py

Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -252,6 +252,17 @@ def test_Block(docproto):
252252

253253
assert block.paragraphs
254254

255+
# Check confidence value
256+
assert isinstance(block.confidence, float)
257+
assert 0.0 <= block.confidence <= 1.0
258+
259+
# Check detected languages
260+
assert isinstance(block.detected_languages, list)
261+
if block.detected_languages:
262+
for language in block.detected_languages:
263+
assert isinstance(language, documentai.Document.Page.DetectedLanguage)
264+
assert hasattr(language, "language_code")
265+
255266

256267
def test_Paragraph(docproto):
257268
wrapped_page = page.Page(
@@ -268,6 +279,17 @@ def test_Paragraph(docproto):
268279

269280
assert paragraph.lines
270281

282+
# Check confidence value
283+
assert isinstance(paragraph.confidence, float)
284+
assert 0.0 <= paragraph.confidence <= 1.0
285+
286+
# Check detected languages
287+
assert isinstance(paragraph.detected_languages, list)
288+
if paragraph.detected_languages:
289+
for language in paragraph.detected_languages:
290+
assert isinstance(language, documentai.Document.Page.DetectedLanguage)
291+
assert hasattr(language, "language_code")
292+
271293

272294
def test_Line(docproto):
273295
wrapped_page = page.Page(
@@ -284,6 +306,17 @@ def test_Line(docproto):
284306

285307
assert line.tokens
286308

309+
# Check confidence value
310+
assert isinstance(line.confidence, float)
311+
assert 0.0 <= line.confidence <= 1.0
312+
313+
# Check detected languages
314+
assert isinstance(line.detected_languages, list)
315+
if line.detected_languages:
316+
for language in line.detected_languages:
317+
assert isinstance(language, documentai.Document.Page.DetectedLanguage)
318+
assert hasattr(language, "language_code")
319+
287320

288321
def test_Token(docproto):
289322
wrapped_page = page.Page(

0 commit comments

Comments
 (0)