diff --git a/tests/environments/execution/test_dask_k8s_environment.py b/tests/environments/execution/test_dask_k8s_environment.py index afd0e7513667..edd7c695d414 100644 --- a/tests/environments/execution/test_dask_k8s_environment.py +++ b/tests/environments/execution/test_dask_k8s_environment.py @@ -341,7 +341,7 @@ def test_populate_worker_pod_yaml_with_private_registry(): ): yaml_obj = environment._populate_worker_pod_yaml(yaml_obj=pod) - yaml_obj["spec"]["imagePullSecrets"][0] == dict(name="foo-man-docker") + assert yaml_obj["spec"]["imagePullSecrets"][0] == dict(name="foo-man-docker") def test_populate_worker_pod_yaml_with_image_pull_secret(): @@ -360,7 +360,7 @@ def test_populate_worker_pod_yaml_with_image_pull_secret(): ): yaml_obj = environment._populate_worker_pod_yaml(yaml_obj=pod) - yaml_obj["spec"]["imagePullSecrets"][0] == dict(name="mysecret") + assert yaml_obj["spec"]["imagePullSecrets"][0] == dict(name="mysecret") def test_initialize_environment_with_spec_populates(monkeypatch): diff --git a/tests/schedules/test_schedules.py b/tests/schedules/test_schedules.py index ebaf1a4e08c2..e2617c850cd3 100644 --- a/tests/schedules/test_schedules.py +++ b/tests/schedules/test_schedules.py @@ -40,7 +40,7 @@ def test_create_schedule_emits_events_if_asked(): assert all([isinstance(e, clocks.ClockEvent) for e in output]) assert all([e.parameter_defaults == dict() for e in output]) - output == [dt.add(days=1), dt.add(days=2), dt.add(days=3)] + assert output == [dt.add(days=1), dt.add(days=2), dt.add(days=3)] def test_create_schedule_multiple_overlapping_clocks():