
    tni
                     r    d Z ddlZddlmZ  G d dej                  Zedk    r ej                     dS dS )z  Test API Wrapper Pod Mutations     N)podsc                   *    e Zd ZdZd Zd Zd Zd ZdS )TestPodMutationszTest API Wrapper Pod Mutationsc                    t          j        d,i ddddddddd	dd
ddddddddddddddddddddiddddd d!d"g}t          j        d#ddddddd$d%dddddid&'          }|                     d(|           |                     d)|           |                     d(|           |                     d*|           d+S )-zQ
        Test generate_pod_deployment_mutation for both GPU and CPU pods
        nametest
image_name
test_imagegpu_type_id1
cloud_typeclouddata_center_idcountry_codeUS	gpu_count   volume_in_gbd   container_disk_in_gb
   min_vcpu_countmin_memory_in_gbdocker_argsargsports8080volume_mount_pathz/pathenvENVsupport_public_ipTtemplate_idabcdeallowed_cuda_versionsz11.8z12.0ztest-cpu      z	cpu3c-2-4)r   r	   r   r   r   r   r   r   r   r   r   r   r   instance_idmutationpodFindAndDeployOnDemanddeployCpuPodN )r    generate_pod_deployment_mutationassertIn)self
gpu_result
cpu_results      _/var/www/html/gpu-tools/venv/lib/python3.11/site-packages/tests/test_api/test_mutations_pods.py%test_generate_pod_deployment_mutationz6TestPodMutations.test_generate_pod_deployment_mutation   s   
 : 
 
 

#|
 
 w	

 3
 
 a
 
 "$
 1
 Q
 
 &
 &g
 
  #d!
"  #
$ $*6"2"2%

, :#!#%#
 
 

$ 	j*---0*=== 	j*---nj11111    c                 Z    t          j        d          }|                     d|           dS )z1
        Test generate_pod_stop_mutation
        pod_idr(   N)r   generate_pod_stop_mutationr-   r.   results     r1   test_generate_pod_stop_mutationz0TestPodMutations.test_generate_pod_stop_mutation?   s.     0::j&)))))r3   c                 \    t          j        dd          }|                     d|           dS )z3
        Test generate_pod_resume_mutation
        r5   r   r(   N)r   generate_pod_resume_mutationr-   r7   s     r1   !test_generate_pod_resume_mutationz2TestPodMutations.test_generate_pod_resume_mutationG   s0     28Q??j&)))))r3   c                 Z    t          j        d          }|                     d|           dS )z6
        Test generate_pod_terminate_mutation
        r5   r(   N)r   generate_pod_terminate_mutationr-   r7   s     r1   $test_generate_pod_terminate_mutationz5TestPodMutations.test_generate_pod_terminate_mutationO   s.     5h??j&)))))r3   N)__name__
__module____qualname____doc__r2   r9   r<   r?   r+   r3   r1   r   r      sW        ((22 22 22h* * ** * ** * * * *r3   r   __main__)rC   unittestrunpod.api.mutationsr   TestCaser   r@   mainr+   r3   r1   <module>rI      s    & &  % % % % % %M* M* M* M* M*x( M* M* M*` zHMOOOOO r3   