o
    i                     @   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_helper5798522612028014592z@projects/964831358985/locations/us-central1/batchPredictionJobs/z,batches/z2p8ksus4lyxt25rntl3fpd67p2niw4hfij5invalid_nametest_get_batch_jobnamezInvalid batch job name)r
   
parametersexception_if_mldevtest_get_batch_operation)r
   r   exception_if_vertex$test_get_batch_job_with_invalid_name)r
   r   r   r   
test_tablezbatches.get)fileglobals_for_filetest_methodr   c                    s4   | j rt}nt}| jjj|dI d H }|sJ d S )Nr	   )vertexai_BATCH_JOB_NAME_MLDEV_BATCH_OPERATION_NAMEaiobatchesget)clientr
   	batch_job r   a/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/google/genai/tests/batches/test_get.pytest_async_getF   s   r   )__doc__pytest r   r   r   _BATCH_JOB_FULL_RESOURCE_NAMEr   _INVALID_BATCH_JOB_NAMETestTableItem_GetBatchJobParametersr   list__annotations__setup__file__globals
pytestmarkmarkasyncior   r   r   r   r   <module>   sT   