o
    Yi,	                     @   s   U d Z ddlZddlmZ ddlmZ dZde Zd	Zd
Zej	dej
eddddej	dej
edddej	dej
eddddgZeej	 ed< ejee dedZejjdd ZdS )zTests for batches.cancel().    N   )types   )pytest_helperZ6339625664542408704z@projects/964831358985/locations/us-central1/batchPredictionJobs/z,batches/0yew7plxupyybd7appsrq5vw7w0lp3l79labZinvalid_nameZtest_cancel_batch_jobnamezInvalid batch job namez2Cannot cancel a batch job multiple times in Vertex)r   
parametersexception_if_mldevZskip_in_api_modeZtest_cancel_batch_operation)r   r   exception_if_vertexZ'test_cancel_batch_job_with_invalid_name)r   r   r	   r
   
test_tablezbatches.cancel)fileZglobals_for_fileZtest_methodr   c                    s,   | j rt}nt}| jjj|dI d H  d S )Nr   )Zvertexai_BATCH_JOB_NAME_MLDEV_BATCH_OPERATION_NAMEZaioZbatchescancel)clientr    r   `/home/kim/smarthome/.venv/lib/python3.10/site-packages/google/genai/tests/batches/test_cancel.pytest_async_cancelG   s
   r   )__doc__Zpytest r   r   r   Z_BATCH_JOB_FULL_RESOURCE_NAMEr   Z_INVALID_BATCH_JOB_NAMEZTestTableItemZ_CancelBatchJobParametersr   list__annotations__setup__file__globalsZ
pytestmarkmarkZasyncior   r   r   r   r   <module>   sV   