Skip to content
This repository has been archived by the owner on Oct 9, 2023. It is now read-only.

Commit

Permalink
upd test
Browse files Browse the repository at this point in the history
Signed-off-by: Miguel <[email protected]>
  • Loading branch information
migueltol22 committed Oct 4, 2021
1 parent 9f28c93 commit b33a9e5
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions go/tasks/plugins/k8s/spark/spark_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -47,6 +47,7 @@ var (
"spark.flyte.feature1.enabled": "true",
"spark.flyteorg.feature2.enabled": "true",
"spark.flyteorg.feature3.enabled": "true",
"spark.batchScheduler": "volcano",
}

dummyEnvVars = []*core.KeyValuePair{
Expand Down Expand Up @@ -384,6 +385,7 @@ func TestBuildResourceSpark(t *testing.T) {
assert.Equal(t, int32(execInstances), *sparkApp.Spec.Executor.Instances)
assert.Equal(t, dummySparkConf["spark.driver.memory"], *sparkApp.Spec.Driver.Memory)
assert.Equal(t, dummySparkConf["spark.executor.memory"], *sparkApp.Spec.Executor.Memory)
assert.Equal(t, dummySparkConf["spark.batchScheduler"], *sparkApp.Spec.BatchScheduler)

// Validate Interruptible Toleration and NodeSelector set for Executor but not Driver.
assert.Equal(t, 0, len(sparkApp.Spec.Driver.Tolerations))
Expand Down

0 comments on commit b33a9e5

Please sign in to comment.