testutil

package
v0.5.3 Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: Jun 3, 2019 License: Apache-2.0 Imports: 11 Imported by: 0

Documentation

Index

Constants

View Source
const (
	TestImageName = "test-image-for-kubeflow-tf-operator:latest"
	TestTFJobName = "test-tfjob"
	LabelWorker   = "worker"
	LabelPS       = "ps"

	SleepInterval = 500 * time.Millisecond
	ThreadCount   = 1
)
View Source
const (
	LabelGroupName = "group-name"
	LabelTFJobName = "tf-job-name"
)

Variables

View Source
var (
	// KeyFunc is the short name to DeletionHandlingMetaNamespaceKeyFunc.
	// IndexerInformer uses a delta queue, therefore for deletes we have to use this
	// key function but it should be just fine for non delete events.
	KeyFunc   = cache.DeletionHandlingMetaNamespaceKeyFunc
	GroupName = tfv1beta2.GroupName
)
View Source
var (
	AlwaysReady = func() bool { return true }
)

Functions

func CheckCondition

func CheckCondition(tfJob *tfv1beta2.TFJob, condition common.JobConditionType, reason string) bool

func ConvertTFJobToUnstructured

func ConvertTFJobToUnstructured(tfJob *tfv1beta2.TFJob) (*unstructured.Unstructured, error)

ConvertTFJobToUnstructured uses JSON to convert TFJob to Unstructured.

func GenLabels

func GenLabels(jobName string) map[string]string

func GenOwnerReference

func GenOwnerReference(tfjob *tfv1beta2.TFJob) *metav1.OwnerReference

func GetKey

func GetKey(tfJob *tfv1beta2.TFJob, t *testing.T) string

func NewBasePod

func NewBasePod(name string, tfJob *tfv1beta2.TFJob, t *testing.T) *v1.Pod

func NewBaseService

func NewBaseService(name string, tfJob *tfv1beta2.TFJob, t *testing.T) *v1.Service

func NewPod

func NewPod(tfJob *tfv1beta2.TFJob, typ string, index int, t *testing.T) *v1.Pod

func NewPodList

func NewPodList(count int32, status v1.PodPhase, tfJob *tfv1beta2.TFJob, typ string, start int32, t *testing.T) []*v1.Pod

create count pods with the given phase for the given tfJob

func NewService

func NewService(tfJob *tfv1beta2.TFJob, typ string, index int, t *testing.T) *v1.Service

func NewServiceList

func NewServiceList(count int32, tfJob *tfv1beta2.TFJob, typ string, t *testing.T) []*v1.Service

NewServiceList creates count pods with the given phase for the given tfJob

func NewTFJob

func NewTFJob(worker, ps int) *tfv1beta2.TFJob

func NewTFJobWithActiveDeadlineSeconds added in v0.5.0

func NewTFJobWithActiveDeadlineSeconds(chief, worker, ps int, ads *int64) *tfv1beta2.TFJob

func NewTFJobWithBackoffLimit added in v0.5.0

func NewTFJobWithBackoffLimit(chief, worker, ps int, backoffLimit *int32) *tfv1beta2.TFJob

func NewTFJobWithChief

func NewTFJobWithChief(worker, ps int) *tfv1beta2.TFJob

func NewTFJobWithCleanPolicy

func NewTFJobWithCleanPolicy(chief, worker, ps int, policy common.CleanPodPolicy) *tfv1beta2.TFJob

func NewTFJobWithCleanupJobDelay

func NewTFJobWithCleanupJobDelay(chief, worker, ps int, ttl *int32) *tfv1beta2.TFJob

func NewTFJobWithEvaluator

func NewTFJobWithEvaluator(worker, ps, evaluator int) *tfv1beta2.TFJob

func NewTFJobWithEvaluatorAndNamespace

func NewTFJobWithEvaluatorAndNamespace(worker, ps, evaluator int, ns string) *tfv1beta2.TFJob

func NewTFJobWithNamespace

func NewTFJobWithNamespace(worker, ps int, ns string) *tfv1beta2.TFJob

func NewTFReplicaSpecTemplate

func NewTFReplicaSpecTemplate() v1.PodTemplateSpec

func SetPodsStatuses

func SetPodsStatuses(podIndexer cache.Indexer, tfJob *tfv1beta2.TFJob, typ string, pendingPods, activePods, succeededPods, failedPods int32, restartCounts []int32, t *testing.T)

func SetServices

func SetServices(serviceIndexer cache.Indexer, tfJob *tfv1beta2.TFJob, typ string, activeWorkerServices int32, t *testing.T)

func SetTFJobCompletionTime

func SetTFJobCompletionTime(tfJob *tfv1beta2.TFJob)

Types

This section is empty.

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL