use decoy better

Opentrons · Mar 14, 2024 · 459d23e · 459d23e
1 parent 2f37b7a
commit 459d23e
Showing 1 changed file with 11 additions and 11 deletions.
diff --git a/api/tests/opentrons/protocol_engine/execution/test_pipetting_handler.py b/api/tests/opentrons/protocol_engine/execution/test_pipetting_handler.py
@@ -502,17 +502,6 @@ async def test_aspirate_volume_validation(
     not_ok_volume = 1.01
     expected_adjusted_volume = 1
 
-    decoy.when(
-        mock_command_note_adder(
-            cast(
-                CommandNote,
-                CommandNoteMatcher(
-                    matching_noteKind_regex="warning",
-                    matching_shortMessage_regex="Aspirate clamped to 1 µL",
-                ),
-            )
-        )
-    ).then_return(None)
     for subject in [virtual_subject, hardware_subject]:
         assert (
             await subject.aspirate_in_place(
@@ -523,6 +512,17 @@ async def test_aspirate_volume_validation(
             )
             == expected_adjusted_volume
         )
+        decoy.verify(
+            mock_command_note_adder(
+                cast(
+                    CommandNote,
+                    CommandNoteMatcher(
+                        matching_noteKind_regex="warning",
+                        matching_shortMessage_regex="Aspirate clamped to 1 µL",
+                    ),
+                )
+            )
+        )
         with pytest.raises(InvalidAspirateVolumeError):
             await subject.aspirate_in_place(
                 pipette_id="pipette-id",