[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

test_preprocess_image.py

@@ -1,6 +1,10 @@
+from model_optimization import \
+    preprocess_image  # Replace with actual module name
+import unittest
 import numpy as np
 from tensorflow.keras.preprocessing.image import img_to_array, load_img
 
+
 def preprocess_image(image_path):
     """Preprocess the image for model prediction."""
     try:
@@ -12,13 +16,14 @@ def preprocess_image(image_path):
 
     image = img_to_array(image)
     if image.shape != (224, 224, 3):
-        raise ValueError(f"Image shape is invalid. Expected (224, 224, 3), got {image.shape}.")
+        raise ValueError(
+            f"Image shape is invalid. Expected (224, 224, 3), got {image.shape}."
+        )
 
     image = image / 255.0  # Normalize
     image = np.expand_dims(image, axis=0)
     return image
-import unittest
-from model_optimization import preprocess_image  # Replace with actual module name
+
 
 class TestPreprocessImage(unittest.TestCase):
     def test_nonexistent_image_path(self):
@@ -37,5 +42,6 @@ def test_invalid_image_shape(self):
         with self.assertRaises(ValueError):
             preprocess_image("test_images/grayscale_image.jpg")
 
+
 if __name__ == "__main__":
     unittest.main()