def METHOD_NAME(self, command: base_command.PartialCommand[typing.Any]):
    mock_check = mock.Mock()
    add_check = mock.Mock()
    command = stub_class(base_command.PartialCommand, add_check=add_check)
    assert command.with_check(mock_check) is mock_check
    add_check.assert_called_once_with(mock_check)