fix: fmt

Author: robgee86

src/arduino/app_peripherals/camera/base_camera.py

@@ -273,16 +273,17 @@ def record_avi(self, duration) -> np.ndarray:
         try:
             fourcc = cv2.VideoWriter.fourcc(*"MJPG")
             out = cv2.VideoWriter(filename, fourcc, self.fps, (width, height))
+
             frame = first_frame
             for i in range(total_frames):
                 if frame is not None:
                     if frame.dtype != np.uint8:
                         frame = _to_uint8(frame)
                     out.write(frame)
-                
+
                 if i < total_frames - 1:
                     frame = self.capture()
-            
+
             out.release()
             with open(filename, "rb") as f:
                 avi_data = f.read()
@@ -410,9 +411,9 @@ def _to_uint8(frame) -> np.ndarray:
     if np.issubdtype(frame.dtype, np.floating):
         # We adopt the OpenCV convention: float images are in [0, 1]
         frame = np.clip(frame * 255, 0, 255)
-    
+
     elif np.issubdtype(frame.dtype, np.integer) and frame.dtype != np.uint8:
         info = np.iinfo(frame.dtype)
         frame = (frame.astype(np.float32) - info.min) / (info.max - info.min) * 255
-    
+
     return frame.astype(np.uint8)

tests/arduino/app_peripherals/camera/test_base_camera.py

@@ -419,30 +419,33 @@ def event_callback(event, data):
     assert "streaming" in events[3][0]
     assert "disconnected" in events[4][0]
 
+
 def test_record_zero_duration():
     camera = MockedCamera()
     camera.start()
     with pytest.raises(ValueError):
         camera.record(0)
     camera.stop()
 
+
 def test_record():
     camera = MockedCamera(fps=5)
     camera.frame = np.ones((480, 640, 3), dtype=np.uint8)
     camera.start()
-    
+
     duration = 1.0
     expected_frames = int(camera.fps * duration)
     frames = camera.record(duration)
-    
+
     assert isinstance(frames, np.ndarray)
     assert frames.shape[0] == expected_frames
     assert frames.shape[1:] == camera.frame.shape
     assert frames.dtype == camera.frame.dtype
     assert np.all(frames == camera.frame)
-    
+
     camera.stop()
 
+
 def test_record_avi():
     camera = MockedCamera(fps=5)
     camera.frame = np.ones((480, 640, 3), dtype=np.uint8)
@@ -455,8 +458,8 @@ def test_record_avi():
     assert isinstance(avi_bytes, np.ndarray)
     assert avi_bytes.dtype == np.uint8
     assert avi_bytes.size > 0
-    
-    with tempfile.NamedTemporaryFile(suffix='.avi') as tmp:
+
+    with tempfile.NamedTemporaryFile(suffix=".avi") as tmp:
         tmp.write(avi_bytes.tobytes())
         tmp.flush()
 
@@ -469,13 +472,14 @@ def test_record_avi():
             assert frame is not None
             assert frame.dtype == np.uint8
             read_count += 1
-        
+
         cap.release()
-    
+
     assert read_count == expected_frames
-    
+
     camera.stop()
 
+
 def test_record_avi_uint8_conversion():
     camera = MockedCamera(fps=5)
     # Use float32 frame, should be converted to uint8 in AVI
@@ -485,15 +489,15 @@ def test_record_avi_uint8_conversion():
     duration = 1.0
     expected_frames = int(camera.fps * duration)
     avi_bytes = camera.record_avi(duration)
-    
+
     assert isinstance(avi_bytes, np.ndarray)
     assert avi_bytes.dtype == np.uint8
     assert avi_bytes.size > 0
-    
-    with tempfile.NamedTemporaryFile(suffix='.avi') as tmp:
+
+    with tempfile.NamedTemporaryFile(suffix=".avi") as tmp:
         tmp.write(avi_bytes.tobytes())
         tmp.flush()
-        
+
         read_count = 0
         cap = cv2.VideoCapture(tmp.name)
         while True:
@@ -503,9 +507,9 @@ def test_record_avi_uint8_conversion():
             assert frame is not None
             assert frame.dtype == np.uint8
             read_count += 1
-        
+
         cap.release()
 
     assert read_count == expected_frames
-    
+
     camera.stop()