style: standardise opencv2 imports to be import cv2 as cv

Author: anna-singleton-resolver

src/resolver_athena_client/client/transformers/core.py

@@ -9,7 +9,7 @@
 import enum
 
 import brotli
-import cv2
+import cv2 as cv
 import numpy as np
 
 from resolver_athena_client.client.consts import EXPECTED_HEIGHT, EXPECTED_WIDTH
@@ -28,10 +28,10 @@ class OpenCVResamplingAlgorithm(enum.Enum):
     resampling algorithms.
     """
 
-    NEAREST = cv2.INTER_NEAREST
-    BOX = cv2.INTER_AREA
-    BILINEAR = cv2.INTER_LINEAR
-    LANCZOS = cv2.INTER_LANCZOS4
+    NEAREST = cv.INTER_NEAREST
+    BOX = cv.INTER_AREA
+    BILINEAR = cv.INTER_LINEAR
+    LANCZOS = cv.INTER_LANCZOS4
 
 
 def _is_raw_bgr_expected_size(data: bytes) -> bool:
@@ -66,7 +66,7 @@ def process_image() -> tuple[bytes, bool]:
 
         # Try to load the image data directly
         img_data_buf = np.frombuffer(image_data.data, dtype=np.uint8)
-        img = cv2.imdecode(img_data_buf, cv2.IMREAD_COLOR)
+        img = cv.imdecode(img_data_buf, cv.IMREAD_COLOR)
 
         if img is None:
             err = "Failed to decode image data for resizing"
@@ -75,7 +75,7 @@ def process_image() -> tuple[bytes, bool]:
         if img.shape[0] == EXPECTED_HEIGHT and img.shape[1] == EXPECTED_WIDTH:
             resized_img = img
         else:
-            resized_img = cv2.resize(
+            resized_img = cv.resize(
                 img, _target_size, interpolation=sampling_algorithm.value
             )
 

tests/client/transformers/test_hash_pipeline.py

@@ -2,7 +2,7 @@
 
 import hashlib
 
-import cv2
+import cv2 as cv
 import numpy as np
 import pytest
 
@@ -27,7 +27,7 @@ def create_test_png_image(width: int = 200, height: int = 200) -> bytes:
     img[:] = (255, 0, 0)  # Red color
 
     # Encode image as PNG to memory
-    success, buffer = cv2.imencode(".png", img)
+    success, buffer = cv.imencode(".png", img)
     if not success:
         err = "Failed to encode image as PNG"
         raise RuntimeError(err)

tests/functional/conftest.py

@@ -1,7 +1,7 @@
 import os
 import uuid
 
-import cv2
+import cv2 as cv
 import numpy as np
 import pytest
 import pytest_asyncio
@@ -26,7 +26,7 @@ def _create_base_test_image_opencv(width: int, height: int) -> np.ndarray:
         height: Image height in pixels
 
     Returns:
-        numpy array in BGR format suitable for cv2.imencode
+        numpy array in BGR format suitable for cv.imencode
     """
     # Create a simple test image with random colors
     # Background color (blue-green)
@@ -36,7 +36,7 @@ def _create_base_test_image_opencv(width: int, height: int) -> np.ndarray:
     # Add an accent rectangle for visual variation
     x1, y1 = width // 4, height // 4
     x2, y2 = (width * 3) // 4, (height * 3) // 4
-    return cv2.rectangle(img_bgr, (x1, y1), (x2, y2), (200, 100, 50), -1)
+    return cv.rectangle(img_bgr, (x1, y1), (x2, y2), (200, 100, 50), -1)
 
 
 SUPPORTED_TEST_FORMATS = [
@@ -138,10 +138,10 @@ def valid_formatted_image(
     # Encode image in the target format
     if image_format in ["pgm", "pbm"]:
         # PGM and PBM are grayscale, so convert the image to grayscale
-        gray_image = cv2.cvtColor(base_image, cv2.COLOR_BGR2GRAY)
-        success, encoded = cv2.imencode(f".{image_format}", gray_image)
+        gray_image = cv.cvtColor(base_image, cv.COLOR_BGR2GRAY)
+        success, encoded = cv.imencode(f".{image_format}", gray_image)
     else:
-        success, encoded = cv2.imencode(f".{image_format}", base_image)
+        success, encoded = cv.imencode(f".{image_format}", base_image)
 
     if not success:
         pytest.fail(f"OpenCV failed to encode image in {image_format} format")