77 lines
1.9 KiB
Python
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",
|
|
],
|
|
)
|