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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.get().    N   )types   )pytest_helperZ5798522612028014592z@projects/964831358985/locations/us-central1/batchPredictionJobs/z,batches/z2p8ksus4lyxt25rntl3fpd67p2niw4hfij5Zinvalid_nameZtest_get_batch_jobnamezInvalid batch job name)r   
parametersexception_if_mldevZtest_get_batch_operation)r   r   exception_if_vertexZ$test_get_batch_job_with_invalid_name)r   r   r	   r
   
test_tablezbatches.get)fileZglobals_for_fileZtest_methodr   c                    s4   | j rt}nt}| jjj|dI d H }|sJ d S )Nr   )Zvertexai_BATCH_JOB_NAME_MLDEV_BATCH_OPERATION_NAMEZaioZbatchesget)clientr   Z	batch_job r   ]/home/kim/smarthome/.venv/lib/python3.10/site-packages/google/genai/tests/batches/test_get.pytest_async_getF   s   r   )__doc__Zpytest r   r   r   Z_BATCH_JOB_FULL_RESOURCE_NAMEr   Z_INVALID_BATCH_JOB_NAMEZTestTableItemZ_GetBatchJobParametersr   list__annotations__setup__file__globalsZ
pytestmarkmarkZasyncior   r   r   r   r   <module>   sT   