load("@rules_python//python:defs.bzl", "py_library")
load("//py/private:pytest.bzl", "pytest_test")
def _is_test(file):
return file.startswith("test_") or file.endswith("_tests.py")
def _suite_suffix(name):
return name[len("test-"):] if name.startswith("test-") else name
def _strip_test_prefixes(path):
if path.endswith(".py"):
path = path[:-len(".py")]
filename = path.rsplit("/", 1)[-1]
if filename.startswith("test_"):
path = path[:-len(filename)] + filename[len("test_"):]
return path
def py_test_suite(name, srcs, size = None, deps = None, python_version = None, imports = None, visibility = None, **kwargs):
library_name = "%s-test-lib" % name
py_library(
name = library_name,
testonly = True,
srcs = srcs,
deps = deps,
imports = imports,
precompile = "disabled",
)
tests = []
for src in srcs:
if _is_test(src):
test_name = "%s-%s" % (_strip_test_prefixes(src), _suite_suffix(name))
tests.append(test_name)
pytest_test(
name = test_name,
size = size,
srcs = [src],
deps = [library_name],
python_version = python_version,
precompile = "disabled",
**kwargs
)
native.test_suite(
name = name,
tests = tests,
visibility = visibility,
)
