fix: fix mypy issues

openfoodfacts · Oct 31, 2023 · 7bf9f67 · 7bf9f67
1 parent 6506d6d
commit 7bf9f67
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/test_ocr.py b/tests/test_ocr.py
@@ -1,4 +1,4 @@
-from typing import Optional
+from typing import Optional, Tuple
 
 import pytest
 
@@ -37,9 +37,10 @@
     ],
 )
 def test_get_words_in_area(
-    ocr_url: str, bounding_box: list[int, int, int, int], expected_text: Optional[str]
+    ocr_url: str, bounding_box: Tuple[int, int, int, int], expected_text: Optional[str]
 ):
     ocr_result = OCRResult.from_url(ocr_url)
+    assert ocr_result is not None
     words = ocr_result.get_words_in_area(bounding_box)
 
     if expected_text is None: