
    R-j>
                        U d Z ddl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       ej                  d ej                  e      d       ej                  d ej                  e	      dd      gZeej                     ed<    ej                  e e       de      Zej&                  j(                  d        Zej&                  j(                  d        Zy)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                    K   | j                   rt        }nt        }| j                  j                  j                  |       d {   }|sJ y 7 	w)Nr
   )vertexai_BATCH_JOB_NAME_MLDEV_BATCH_OPERATION_NAMEaiobatchesgetclientr   	batch_jobs      q/Users/jameslopez/projects/TradingBot25/.venv/lib/python3.12/site-packages/google/genai/tests/batches/test_get.pytest_async_getr    G   sD     __D&DJJ&&***55)	 6s   AAA
Ac                    K   | j                   rat        }| j                  j                  j	                  |       d {   }t        j                  d|j                  j                        sJ y y 7 2w)Nr
   z/^projects/[^/]+/locations/[^/]+/datasets/[^/]+$)	r   r   r   r   r   rematchoutput_infovertex_multimodal_dataset_namer   s      r   -test_async_get_with_multimodal_dataset_outputr&   R   sb     __Djj((,,$,77I88:<<  	 7s   <A3A13A3)__doc__r"   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   <module>r8      sB  "  	    (*+- 
 M ( 
  M!/5// 
 4  M'/5//,
 5  M3/5//(
 441
D,,- 2 !]  	Y	
    r7   