Adds unit test for single sample ocr with GoogleOCR

CHANGELOG.md

@@ -13,6 +13,7 @@ The version numbers are according to [Semantic Versioning](http://semver.org/).
 ### Fixed
 - Adds forced conversion to RGB in pillow before sending data to OpenCV to fix a possible bug in Studio
 - Fixes a rare bug where self-intersecting bounding boxes caused the OCR system to crash when using multi-pass OCR
+
 ### Removed
 
 

tests/test_googleocr.py

@@ -45,7 +45,15 @@ def unrotated_bboxes(ocr):
 
 def test_google_ocr(ocr):
     img = Image.open(os.path.join(DATA_DIR, "ocr_test_big.png"))
+    res = ocr.ocr(img)
+    text = " ".join([r["text"] for r in res])
+    assert text == "This is a test ."
+    assert all([r["bbox"].original_size == img.size for r in res])
+
 
+def test_google_orc_single_sample():
+    img = Image.open(os.path.join(DATA_DIR, "ocr_test_big.png"))
+    ocr = GoogleOCR(auto_rotate=True, ocr_samples=1)
     res = ocr.ocr(img)
     text = " ".join([r["text"] for r in res])
     assert text == "This is a test ."