Files
2026-07-13 13:17:40 +08:00

77 lines
1.9 KiB
Python

load("@rules_python//python:defs.bzl", "py_library")
# --------------------------------------------------------------------
# Tests from the python/ray/autoscaler/v2/tests directory.
# Covers all tests starting with `test_`.
# Please keep these sorted alphabetically.
# --------------------------------------------------------------------
load("//bazel:python.bzl", "py_test_module_list")
py_library(
name = "conftest",
srcs = ["tests/conftest.py"],
deps = ["//python/ray/tests:conftest"],
)
# Unit test. (These should not have ray started)
py_test_module_list(
size = "small",
files = [
"tests/test_allocation_timeout_recovery.py",
"tests/test_config.py",
"tests/test_event_logger.py",
"tests/test_instance_manager.py",
"tests/test_instance_storage.py",
"tests/test_instance_util.py",
"tests/test_ippr_provider.py",
"tests/test_metrics_reporter.py",
"tests/test_node_provider.py",
"tests/test_priority_selection.py",
"tests/test_ray_installer.py",
"tests/test_reconciler.py",
"tests/test_scheduler.py",
"tests/test_schema.py",
"tests/test_storage.py",
"tests/test_subscribers.py",
"tests/test_threaded_ray_installer.py",
"tests/test_utils.py",
],
tags = ["team:core"],
deps = [
":conftest",
"//:ray_lib",
],
)
# Integration test.
py_test_module_list(
size = "medium",
files = [
"tests/test_autoscaler.py",
"tests/test_sdk.py",
],
tags = [
"exclusive",
"team:core",
],
deps = [
":conftest",
"//:ray_lib",
],
)
py_test_module_list(
size = "large",
files = [
"tests/test_e2e.py",
],
tags = [
"exclusive",
"team:core",
],
deps = [
":conftest",
"//:ray_lib",
],
)