Skip to content

Commit 5d83b1d

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 7396b8a commit 5d83b1d

File tree

2 files changed

+21
-18
lines changed

2 files changed

+21
-18
lines changed

tests/model_registry/model_catalog/test_default_model_catalog.py

Lines changed: 13 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -10,8 +10,12 @@
1010
from ocp_resources.config_map import ConfigMap
1111
from ocp_resources.route import Route
1212
from ocp_resources.service import Service
13-
from tests.model_registry.model_catalog.utils import validate_model_catalog_enabled, execute_get_command, \
14-
validate_model_catalog_resource, validate_default_catalog
13+
from tests.model_registry.model_catalog.utils import (
14+
validate_model_catalog_enabled,
15+
execute_get_command,
16+
validate_model_catalog_resource,
17+
validate_default_catalog,
18+
)
1519

1620
LOGGER = get_logger(name=__name__)
1721

@@ -21,7 +25,6 @@
2125
"model_registry_namespace",
2226
)
2327
class TestModelCatalog:
24-
2528
def test_config_map_exists(self: Self, catalog_config_map: ConfigMap):
2629
# Check that the default configmaps is created when model registry is
2730
# enabled on data science cluster.
@@ -52,14 +55,14 @@ def test_config_map_exists(self: Self, catalog_config_map: ConfigMap):
5255
),
5356
],
5457
)
55-
def test_model_catalog_resources_exists(self: Self, admin_client: DynamicClient, model_registry_namespace: str,
56-
resource_name: Any):
57-
validate_model_catalog_resource(kind=resource_name, admin_client=admin_client,
58-
namespace=model_registry_namespace)
59-
60-
def test_operator_pod_enabled_model_catalog(
61-
self: Self, model_registry_operator_pod: Pod
58+
def test_model_catalog_resources_exists(
59+
self: Self, admin_client: DynamicClient, model_registry_namespace: str, resource_name: Any
6260
):
61+
validate_model_catalog_resource(
62+
kind=resource_name, admin_client=admin_client, namespace=model_registry_namespace
63+
)
64+
65+
def test_operator_pod_enabled_model_catalog(self: Self, model_registry_operator_pod: Pod):
6366
assert validate_model_catalog_enabled(pod=model_registry_operator_pod)
6467

6568
def test_model_catalog_no_custom_catalog(

tests/model_registry/model_catalog/utils.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -62,15 +62,15 @@ def wait_for_model_catalog_update(client: DynamicClient, model_registry_namespac
6262
return True
6363
return False
6464

65+
6566
def validate_model_catalog_resource(kind: Any, admin_client: DynamicClient, namespace: str) -> None:
66-
resource = list(kind.get(
67-
namespace=namespace, label_selector="component=model-catalog", dyn_client=admin_client
68-
))
67+
resource = list(kind.get(namespace=namespace, label_selector="component=model-catalog", dyn_client=admin_client))
6968
assert resource
70-
assert len(resource) == 1, f"Unexpected number of {kind} resources found: {[res.name for res in resource]}"
69+
assert len(resource) == 1, f"Unexpected number of {kind} resources found: {[res.name for res in resource]}"
70+
7171

7272
def validate_default_catalog(default_catalog) -> None:
73-
assert default_catalog["name"] == 'Default Catalog'
74-
assert default_catalog['id'] == 'default_catalog'
75-
assert default_catalog['type'] == 'yaml'
76-
assert default_catalog['properties'].get("yamlCatalogPath") == '/default-catalog.yaml'
73+
assert default_catalog["name"] == "Default Catalog"
74+
assert default_catalog["id"] == "default_catalog"
75+
assert default_catalog["type"] == "yaml"
76+
assert default_catalog["properties"].get("yamlCatalogPath") == "/default-catalog.yaml"

0 commit comments

Comments
 (0)