Update hardware tests and fix PNG saving with picamera

tests/hardware_specific_tests/picamera2/test_acquisition.py

@@ -7,11 +7,8 @@ import numpy as np
 from PIL import Image
 
 
-def test_jpeg_and_array(picamera_client):
-    """Check that a jpeg grabbed from the stream is the same size as other captures.
-
-    Compare it to an array capture and a jpeg capture.
-    """
+def test_quick_capture_size(picamera_client):
+    """Check that a jpeg grabbed from the stream is the same size as a quick capture."""
     # Grab a jpeg from the stream
     blob = picamera_client.grab_jpeg()
     mjpeg_frame = Image.open(blob.open())
@@ -20,13 +17,17 @@ def test_jpeg_and_array(picamera_client):
     assert mjpeg_frame.format == "JPEG"
 
     # Capture a jpeg
-    blob = picamera_client.capture_jpeg(stream_name="main")
+    blob = picamera_client.capture(
+        capture_mode="quick",
+        image_format="jpeg",
+        retain_image=True,
+    )
     jpeg_capture = Image.open(blob.open())
     jpeg_capture.verify()
     assert jpeg_capture.format == "JPEG"
 
     # Capture an array
-    arrlist = picamera_client.capture_as_array(stream_name="main")
+    arrlist = picamera_client.capture_as_array(capture_mode="quick")
     array_main = np.array(arrlist)
 
     # Verify image sizes are the same
@@ -34,6 +35,41 @@ def test_jpeg_and_array(picamera_client):
     assert array_main.shape[1::-1] == jpeg_capture.size
 
 
+def test_format(picamera_client):
+    """Check capture format is as requested."""
+    # Capture a jpeg
+    blob = picamera_client.capture(
+        capture_mode="quick",
+        image_format="jpeg",
+        retain_image=True,
+    )
+    jpeg_capture = Image.open(blob.open())
+    jpeg_capture.verify()
+    assert jpeg_capture.format == "JPEG"
+
+    blob = picamera_client.capture(
+        capture_mode="quick",
+        image_format="png",
+        retain_image=True,
+    )
+    jpeg_capture = Image.open(blob.open())
+    jpeg_capture.verify()
+    assert jpeg_capture.format == "PNG"
+
+
+def test_standard_capture_size(picamera_client):
+    """Check standard capture mode captures at expected size."""
+    # Capture a jpeg
+    blob = picamera_client.capture(
+        capture_mode="standard",
+        image_format="jpeg",
+        retain_image=True,
+    )
+    jpeg_capture = Image.open(blob.open())
+    jpeg_capture.verify()
+    assert jpeg_capture.size == (1640, 1232)
+
+
 def test_record_framerate(picamera_client):
     """Check that framerate monitoring creates a valid JSON log with good data."""
     log_file = Path(picamera_client.record_framerate(duration=1.0))

tests/hardware_specific_tests/picamera2/test_exposure_time_drift.py

@@ -39,7 +39,7 @@ def _test_exposure_time_drift(desired_time: int) -> None:
         assert abs(pre_capture_et - desired_time) < EXPOSURE_TOL
 
         for i in range(10):
-            client.capture_jpeg(stream_name="full")
+            client.capture(capture_mode="full")
             if i == 0:
                 # Exposure can update on first capture, due to frame rate restrictions
                 first_et = client.exposure_time
@@ -56,7 +56,7 @@ def _test_exposure_time_drift(desired_time: int) -> None:
         time.sleep(0.5)
         # Check before and after capture
         assert client.exposure_time == frame_et
-        client.capture_jpeg(stream_name="full")
+        client.capture(capture_mode="full")
         assert client.exposure_time == frame_et
         print("Exposure time didn't change!!")
     print(f"End of test for exposure target {desired_time}")
@@ -82,7 +82,7 @@ def test_exposure_time_on_start_and_stop_stream():
         # Take a couple of images to make sure that the exposure is adjusted to
         # a hardware compatible value.
         for _i in range(2):
-            client.capture_jpeg(stream_name="full")
+            client.capture(capture_mode="full")
         # Save this time.
         set_time = client.exposure_time
         assert abs(set_time - desired_time) < EXPOSURE_TOL
@@ -112,7 +112,7 @@ def _load_camera_and_return_exposure(tmpdir: str) -> int:
         # Take a couple of images to make sure that the exposure is adjusted to
         # a hardware compatible value.
         for _i in range(2):
-            client.capture_jpeg(stream_name="full")
+            client.capture(capture_mode="full")
         # Save this time.
         return client.exposure_time
 

tests/hardware_specific_tests/picamera2/test_streaming_mode.py

@@ -14,7 +14,7 @@ def test_streaming_mode():
     with camera_test_client() as client:
         for mode, res in ["default", (820, 616)], ["full_resolution", (3280, 2464)]:
             client.change_streaming_mode(mode=mode)
-            arr = np.array(client.capture_as_array(stream_name="main"))
+            arr = np.array(client.capture_as_array(capture_mode="quick"))
             # Check that the array dimensions match the requested image size.
             # Note: Numpy array shape is (y,x), but the sensor is set with (x,y)
             # hence the need to compare index 0 with index 1.