def test_dual_transform(image, mask):
image_call = call(image, interpolation=cv2.INTER_LINEAR, cols=image.shape[1], rows=image.shape[0])
mask_call = call(mask, interpolation=cv2.INTER_NEAREST, cols=mask.shape[1], rows=mask.shape[0])
with mock.patch.object(DualTransform, "apply") as mocked_apply:
with mock.patch.object(DualTransform, "get_params", return_value={"interpolation": cv2.INTER_LINEAR}):
aug = DualTransform(p=1)
aug(image=image, mask=mask)
mocked_apply.assert_has_calls([image_call, mask_call], any_order=True)
def test_additional_targets(image, mask):
image_call = call(image,
interpolation=cv2.INTER_LINEAR,
cols=image.shape[1],
rows=image.shape[0])
image2_call = call(mask,
interpolation=cv2.INTER_LINEAR,
cols=mask.shape[1],
rows=mask.shape[0])
with mock.patch.object(DualTransform, 'apply') as mocked_apply:
with mock.patch.object(
DualTransform,
'get_params',
return_value={'interpolation': cv2.INTER_LINEAR}):
aug = DualTransform(p=1)
aug.add_targets({'image2': 'image'})
aug(image=image, image2=mask)
mocked_apply.assert_has_calls([image_call, image2_call],
any_order=True)
