28
28
from opentrons .protocol_engine .resources .model_utils import ModelUtils
29
29
from opentrons .protocol_engine .state import update_types
30
30
from opentrons .protocol_engine .state .state import StateView
31
- from opentrons .protocol_engine .execution import MovementHandler , TipHandler
31
+ from opentrons .protocol_engine .execution import MovementHandler , GantryMover , TipHandler
32
32
33
33
34
34
from opentrons .types import Point
@@ -60,7 +60,7 @@ def mock_model_utils(decoy: Decoy) -> ModelUtils:
60
60
61
61
def test_drop_tip_params_defaults () -> None :
62
62
"""A drop tip should use a `WellOrigin.DROP_TIP` by default."""
63
- default_params = EvotipUnsealPipetteParams .parse_obj (
63
+ default_params = EvotipUnsealPipetteParams .model_validate (
64
64
{"pipetteId" : "abc" , "labwareId" : "def" , "wellName" : "ghj" }
65
65
)
66
66
@@ -71,7 +71,7 @@ def test_drop_tip_params_defaults() -> None:
71
71
72
72
def test_drop_tip_params_default_origin () -> None :
73
73
"""A drop tip should drop a `WellOrigin.DROP_TIP` by default even if an offset is given."""
74
- default_params = EvotipUnsealPipetteParams .parse_obj (
74
+ default_params = EvotipUnsealPipetteParams .model_validate (
75
75
{
76
76
"pipetteId" : "abc" ,
77
77
"labwareId" : "def" ,
@@ -91,21 +91,22 @@ async def test_drop_tip_implementation(
91
91
mock_movement_handler : MovementHandler ,
92
92
mock_tip_handler : TipHandler ,
93
93
mock_model_utils : ModelUtils ,
94
+ gantry_mover : GantryMover ,
94
95
) -> None :
95
96
"""A DropTip command should have an execution implementation."""
96
97
subject = EvotipUnsealPipetteImplementation (
97
98
state_view = mock_state_view ,
98
99
movement = mock_movement_handler ,
99
100
tip_handler = mock_tip_handler ,
100
101
model_utils = mock_model_utils ,
102
+ gantry_mover = gantry_mover ,
101
103
)
102
104
103
105
params = EvotipUnsealPipetteParams (
104
106
pipetteId = "abc" ,
105
107
labwareId = "123" ,
106
108
wellName = "A3" ,
107
109
wellLocation = DropTipWellLocation (offset = WellOffset (x = 1 , y = 2 , z = 3 )),
108
- homeAfter = True ,
109
110
)
110
111
111
112
decoy .when (
@@ -169,21 +170,21 @@ async def test_drop_tip_with_alternating_locations(
169
170
mock_movement_handler : MovementHandler ,
170
171
mock_tip_handler : TipHandler ,
171
172
mock_model_utils : ModelUtils ,
173
+ gantry_mover : GantryMover ,
172
174
) -> None :
173
175
"""It should drop tip at random location within the labware every time."""
174
176
subject = EvotipUnsealPipetteImplementation (
175
177
state_view = mock_state_view ,
176
178
movement = mock_movement_handler ,
177
179
tip_handler = mock_tip_handler ,
178
180
model_utils = mock_model_utils ,
181
+ gantry_mover = gantry_mover ,
179
182
)
180
183
params = EvotipUnsealPipetteParams (
181
184
pipetteId = "abc" ,
182
185
labwareId = "123" ,
183
186
wellName = "A3" ,
184
187
wellLocation = DropTipWellLocation (offset = WellOffset (x = 1 , y = 2 , z = 3 )),
185
- homeAfter = True ,
186
- alternateDropLocation = True ,
187
188
)
188
189
drop_location = DropTipWellLocation (
189
190
origin = DropTipWellOrigin .DEFAULT , offset = WellOffset (x = 10 , y = 20 , z = 30 )
@@ -249,13 +250,15 @@ async def test_tip_attached_error(
249
250
mock_movement_handler : MovementHandler ,
250
251
mock_tip_handler : TipHandler ,
251
252
mock_model_utils : ModelUtils ,
253
+ gantry_mover : GantryMover ,
252
254
) -> None :
253
255
"""A Evotip Unseal command should have an execution implementation."""
254
256
subject = EvotipUnsealPipetteImplementation (
255
257
state_view = mock_state_view ,
256
258
movement = mock_movement_handler ,
257
259
tip_handler = mock_tip_handler ,
258
260
model_utils = mock_model_utils ,
261
+ gantry_mover = gantry_mover ,
259
262
)
260
263
261
264
params = EvotipUnsealPipetteParams (
@@ -303,7 +306,7 @@ async def test_tip_attached_error(
303
306
result = await subject .execute (params )
304
307
305
308
assert result == DefinedErrorData (
306
- public = TipPhysicallyAttachedError . construct (
309
+ public = StallOrCollisionError . model_construct (
307
310
id = "error-id" ,
308
311
createdAt = datetime (year = 1 , month = 2 , day = 3 ),
309
312
wrappedErrors = [matchers .Anything ()],
@@ -345,13 +348,15 @@ async def test_stall_error(
345
348
mock_movement_handler : MovementHandler ,
346
349
mock_tip_handler : TipHandler ,
347
350
mock_model_utils : ModelUtils ,
351
+ gantry_mover : GantryMover ,
348
352
) -> None :
349
353
"""A DropTip command should have an execution implementation."""
350
354
subject = EvotipUnsealPipetteImplementation (
351
355
state_view = mock_state_view ,
352
356
movement = mock_movement_handler ,
353
357
tip_handler = mock_tip_handler ,
354
358
model_utils = mock_model_utils ,
359
+ gantry_mover = gantry_mover ,
355
360
)
356
361
357
362
params = EvotipUnsealPipetteParams (
@@ -396,7 +401,7 @@ async def test_stall_error(
396
401
result = await subject .execute (params )
397
402
398
403
assert result == DefinedErrorData (
399
- public = StallOrCollisionError .construct (
404
+ public = StallOrCollisionError .model_construct (
400
405
id = "error-id" ,
401
406
createdAt = datetime (year = 1 , month = 2 , day = 3 ),
402
407
wrappedErrors = [matchers .Anything ()],
0 commit comments