added unit test

Signed-off-by: Shashank Mittal <[email protected]>
kubeflow · Aug 15, 2024 · f4ca526 · f4ca526
1 parent ff9137b
commit f4ca526
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/pkg/webhook/v1beta1/experiment/validator/validator_test.go b/pkg/webhook/v1beta1/experiment/validator/validator_test.go
@@ -455,6 +455,17 @@ func TestValidateParameters(t *testing.T) {
 			},
 			testDescription: "Not empty max for categorical parameter type",
 		},
+		{
+			parameters: func() []experimentsv1beta1.ParameterSpec {
+				ps := newFakeInstance().Spec.Parameters
+				ps[0].FeasibleSpace.Distribution = "invalid-distribution"
+				return ps
+			}(),
+			wantErr: field.ErrorList{
+				field.Invalid(field.NewPath("spec").Child("parameters").Index(0).Child("feasibleSpace").Child("distribution"), "", ""),
+			},
+			testDescription: "Invalid distribution type",
+		},
 	}
 
 	for _, tc := range tcs {