github.com/kubeflow/training-operator@v1.7.0/sdk/python/test/test_kubeflow_org_v1_mx_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_mx_job import KubeflowOrgV1MXJob # noqa: E501 20 from kubeflow.training.rest import ApiException 21 22 class TestKubeflowOrgV1MXJob(unittest.TestCase): 23 """KubeflowOrgV1MXJob 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 KubeflowOrgV1MXJob 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_mx_job.KubeflowOrgV1MXJob() # noqa: E501 37 if include_optional : 38 return KubeflowOrgV1MXJob( 39 api_version = '0', 40 kind = '0', 41 metadata = None, 42 spec = kubeflow_org_v1_mx_job_spec.KubeflowOrgV1MXJobSpec( 43 job_mode = '0', 44 mx_replica_specs = { 45 'key' : kubeflow_org_v1_replica_spec.KubeflowOrgV1ReplicaSpec( 46 replicas = 56, 47 restart_policy = '0', 48 template = None, ) 49 }, 50 run_policy = kubeflow_org_v1_run_policy.KubeflowOrgV1RunPolicy( 51 active_deadline_seconds = 56, 52 backoff_limit = 56, 53 clean_pod_policy = '0', 54 scheduling_policy = kubeflow_org_v1_scheduling_policy.KubeflowOrgV1SchedulingPolicy( 55 min_available = 56, 56 min_resources = { 57 'key' : None 58 }, 59 priority_class = '0', 60 queue = '0', 61 schedule_timeout_seconds = 56, ), 62 suspend = True, 63 ttl_seconds_after_finished = 56, ), ), 64 status = kubeflow_org_v1_job_status.KubeflowOrgV1JobStatus( 65 completion_time = None, 66 conditions = [ 67 kubeflow_org_v1_job_condition.KubeflowOrgV1JobCondition( 68 last_transition_time = None, 69 last_update_time = None, 70 message = '0', 71 reason = '0', 72 status = '0', 73 type = '0', ) 74 ], 75 last_reconcile_time = None, 76 replica_statuses = { 77 'key' : kubeflow_org_v1_replica_status.KubeflowOrgV1ReplicaStatus( 78 active = 56, 79 failed = 56, 80 label_selector = None, 81 selector = '0', 82 succeeded = 56, ) 83 }, 84 start_time = None, ) 85 ) 86 else : 87 return KubeflowOrgV1MXJob( 88 ) 89 90 def testKubeflowOrgV1MXJob(self): 91 """Test KubeflowOrgV1MXJob""" 92 inst_req_only = self.make_instance(include_optional=False) 93 inst_req_and_optional = self.make_instance(include_optional=True) 94 95 96 if __name__ == '__main__': 97 unittest.main()