def test_default_toleration_preserved(image_name): pod_spec = clean_pod_template( make_pod_spec( image=image_name, extra_pod_config={ "tolerations": [ { "key": "example.org/toleration", "operator": "Exists", "effect": "NoSchedule", } ] }, ) ) tolerations = pod_spec.to_dict()["spec"]["tolerations"] assert { "key": "k8s.dask.org/dedicated", "operator": "Equal", "value": "worker", "effect": "NoSchedule", "toleration_seconds": None, } in tolerations assert { "key": "k8s.dask.org_dedicated", "operator": "Equal", "value": "worker", "effect": "NoSchedule", "toleration_seconds": None, } in tolerations assert { "key": "example.org/toleration", "operator": "Exists", "effect": "NoSchedule", } in tolerations
def pod_spec(docker_image): yield clean_pod_template( make_pod_spec( image=docker_image, extra_container_config={"imagePullPolicy": "IfNotPresent"}, ) )
def pod_spec(image_name): yield clean_pod_template( make_pod_spec( image=image_name, extra_container_config={"imagePullPolicy": "IfNotPresent"} ) )