github.com/kubeflow/training-operator@v1.7.0/sdk/python/test/test_kubeflow_org_v1_mpi_job.py (about)

     1  # coding: utf-8
     2  
     3  """
     4      Kubeflow Training SDK
     5  
     6      Python SDK for Kubeflow Training  # noqa: E501
     7  
     8      The version of the OpenAPI document: v1.7.0
     9      Generated by: https://openapi-generator.tech
    10  """
    11  
    12  
    13  from __future__ import absolute_import
    14  
    15  import unittest
    16  import datetime
    17  
    18  from kubeflow.training.models import *
    19  from kubeflow.training.models.kubeflow_org_v1_mpi_job import KubeflowOrgV1MPIJob  # noqa: E501
    20  from kubeflow.training.rest import ApiException
    21  
    22  class TestKubeflowOrgV1MPIJob(unittest.TestCase):
    23      """KubeflowOrgV1MPIJob unit test stubs"""
    24  
    25      def setUp(self):
    26          pass
    27  
    28      def tearDown(self):
    29          pass
    30  
    31      def make_instance(self, include_optional):
    32          """Test KubeflowOrgV1MPIJob
    33              include_option is a boolean, when False only required
    34              params are included, when True both required and
    35              optional params are included """
    36          # model = kubeflow.training.models.kubeflow_org_v1_mpi_job.KubeflowOrgV1MPIJob()  # noqa: E501
    37          if include_optional :
    38              return KubeflowOrgV1MPIJob(
    39                  api_version = '0', 
    40                  kind = '0', 
    41                  metadata = None, 
    42                  spec = kubeflow_org_v1_mpi_job_spec.KubeflowOrgV1MPIJobSpec(
    43                      clean_pod_policy = '0', 
    44                      main_container = '0', 
    45                      mpi_replica_specs = {
    46                          'key' : kubeflow_org_v1_replica_spec.KubeflowOrgV1ReplicaSpec(
    47                              replicas = 56, 
    48                              restart_policy = '0', 
    49                              template = None, )
    50                          }, 
    51                      run_policy = kubeflow_org_v1_run_policy.KubeflowOrgV1RunPolicy(
    52                          active_deadline_seconds = 56, 
    53                          backoff_limit = 56, 
    54                          clean_pod_policy = '0', 
    55                          scheduling_policy = kubeflow_org_v1_scheduling_policy.KubeflowOrgV1SchedulingPolicy(
    56                              min_available = 56, 
    57                              min_resources = {
    58                                  'key' : None
    59                                  }, 
    60                              priority_class = '0', 
    61                              queue = '0', 
    62                              schedule_timeout_seconds = 56, ), 
    63                          suspend = True, 
    64                          ttl_seconds_after_finished = 56, ), 
    65                      slots_per_worker = 56, ), 
    66                  status = kubeflow_org_v1_job_status.KubeflowOrgV1JobStatus(
    67                      completion_time = None, 
    68                      conditions = [
    69                          kubeflow_org_v1_job_condition.KubeflowOrgV1JobCondition(
    70                              last_transition_time = None, 
    71                              last_update_time = None, 
    72                              message = '0', 
    73                              reason = '0', 
    74                              status = '0', 
    75                              type = '0', )
    76                          ], 
    77                      last_reconcile_time = None, 
    78                      replica_statuses = {
    79                          'key' : kubeflow_org_v1_replica_status.KubeflowOrgV1ReplicaStatus(
    80                              active = 56, 
    81                              failed = 56, 
    82                              label_selector = None, 
    83                              selector = '0', 
    84                              succeeded = 56, )
    85                          }, 
    86                      start_time = None, )
    87              )
    88          else :
    89              return KubeflowOrgV1MPIJob(
    90          )
    91  
    92      def testKubeflowOrgV1MPIJob(self):
    93          """Test KubeflowOrgV1MPIJob"""
    94          inst_req_only = self.make_instance(include_optional=False)
    95          inst_req_and_optional = self.make_instance(include_optional=True)
    96  
    97  
    98  if __name__ == '__main__':
    99      unittest.main()