mirror of
https://github.com/hanzo-ml/kserve.git
synced 2026-10-30 20:53:49 +00:00
778f1fc122
Signed-off-by: greg pereira <[email protected]> Signed-off-by: Vivek Karunai Kiri Ragavan <[email protected]> Co-authored-by: Vivek Karunai Kiri Ragavan <[email protected]>
62 lines
2.3 KiB
Python
62 lines
2.3 KiB
Python
# Copyright 2025 The KServe Authors.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
# Fixture factory - not called explicitly, but must be imported for pytest to discover it.
|
|
from .fixtures import test_case # noqa: F401
|
|
|
|
|
|
# This hook is used to ensure that the test names are unique and to ensure that
|
|
# the test names are consistent with the cluster marks.
|
|
def pytest_collection_modifyitems(config, items):
|
|
for item in items:
|
|
# only touch parameterized tests
|
|
if not hasattr(item, "callspec"):
|
|
continue
|
|
|
|
# if there's no [...] suffix (i.e. not parametrized), skip
|
|
if "[" not in item.nodeid:
|
|
continue
|
|
base, rest = item.nodeid.split("[", 1)
|
|
rest = rest.rstrip("]")
|
|
|
|
cluster_marks = [
|
|
m.name for m in item.iter_markers() if m.name.startswith("cluster_")
|
|
]
|
|
if not cluster_marks:
|
|
continue
|
|
|
|
new_id = "-".join(cluster_marks + [rest])
|
|
item._nodeid = f"{base}[{new_id}]"
|
|
|
|
|
|
def pytest_configure(config):
|
|
config.addinivalue_line(
|
|
"markers", "llminferenceservice: mark test as an LLM inference service test"
|
|
)
|
|
config.addinivalue_line("markers", "autoscaling: mark test as an autoscaling test")
|
|
config.addinivalue_line(
|
|
"markers", "autoscaling_hpa: mark test as an HPA autoscaling test"
|
|
)
|
|
config.addinivalue_line(
|
|
"markers", "autoscaling_keda: mark test as a KEDA autoscaling test"
|
|
)
|
|
config.addinivalue_line(
|
|
"markers", "model_routing: mark test as a model-based routing test"
|
|
)
|
|
config.addinivalue_line("markers", "lora: mark test as a LoRA adapter test")
|
|
config.addinivalue_line("markers", "pvc_storage: mark test as a PVC storage test")
|
|
config.addinivalue_line(
|
|
"markers", "tracing: mark test as a distributed tracing test"
|
|
)
|