|
10 | 10 | from ocp_resources.config_map import ConfigMap |
11 | 11 | from ocp_resources.route import Route |
12 | 12 | 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 | +) |
15 | 19 |
|
16 | 20 | LOGGER = get_logger(name=__name__) |
17 | 21 |
|
|
21 | 25 | "model_registry_namespace", |
22 | 26 | ) |
23 | 27 | class TestModelCatalog: |
24 | | - |
25 | 28 | def test_config_map_exists(self: Self, catalog_config_map: ConfigMap): |
26 | 29 | # Check that the default configmaps is created when model registry is |
27 | 30 | # enabled on data science cluster. |
@@ -52,14 +55,14 @@ def test_config_map_exists(self: Self, catalog_config_map: ConfigMap): |
52 | 55 | ), |
53 | 56 | ], |
54 | 57 | ) |
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 |
62 | 60 | ): |
| 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): |
63 | 66 | assert validate_model_catalog_enabled(pod=model_registry_operator_pod) |
64 | 67 |
|
65 | 68 | def test_model_catalog_no_custom_catalog( |
|
0 commit comments