blob: e637310cd4b2f9216f34d9bdff1070b6fa52110c [file] [log] [blame]
diff --git a/python/internal.bzl b/python/internal.bzl
index 8f99becec..8c667e1ae 100644
--- a/python/internal.bzl
+++ b/python/internal.bzl
@@ -1,6 +1,7 @@
"""
Internal helpers for building the Python protobuf runtime.
"""
+load("@rules_python//python:py_test.bzl", "py_test")
def _remove_cross_repo_path(path):
components = path.split("/")
@@ -123,7 +124,7 @@ def internal_py_test(deps = [], **kwargs):
deps: any additional dependencies of the test.
**kwargs: arguments forwarded to py_test.
"""
- native.py_test(
+ py_test(
imports = ["."],
deps = deps + ["//python:python_test_lib"],
target_compatible_with = select({
diff --git a/python/pb_unit_tests/pyproto_test_wrapper.bzl b/python/pb_unit_tests/pyproto_test_wrapper.bzl
index b5f474d3d..1e2b5daa2 100644
--- a/python/pb_unit_tests/pyproto_test_wrapper.bzl
+++ b/python/pb_unit_tests/pyproto_test_wrapper.bzl
@@ -1,8 +1,10 @@
"""Wrapper for another py_test to run with upb, possibly with a set of expected failures."""
+load("@rules_python//python:py_test.bzl", "py_test")
+
def pyproto_test_wrapper(name, deps = []):
src = name + "_wrapper.py"
- native.py_test(
+ py_test(
name = name,
srcs = [src],
legacy_create_init = False,
diff --git a/python/py_extension.bzl b/python/py_extension.bzl
index 9a2828b95..2741cdf6e 100644
--- a/python/py_extension.bzl
+++ b/python/py_extension.bzl
@@ -1,6 +1,7 @@
"""Macro to support py_extension """
load("@bazel_skylib//lib:selects.bzl", "selects")
+load("@rules_python//python:py_library.bzl", "py_library")
def py_extension(name, srcs, copts, deps = [], **kwargs):
"""Creates a C++ library to extend python
@@ -50,7 +51,7 @@ def py_extension(name, srcs, copts, deps = [], **kwargs):
visibility = ["//python:__subpackages__"],
)
- native.py_library(
+ py_library(
name = name,
data = [output_file],
imports = ["."],
diff --git a/upb/cmake/build_defs.bzl b/upb/cmake/build_defs.bzl
index 325189c7f..c7a7a4558 100644
--- a/upb/cmake/build_defs.bzl
+++ b/upb/cmake/build_defs.bzl
@@ -7,6 +7,8 @@
"""Bazel support functions related to CMake support."""
+load("@rules_python//python:py_test.bzl", "py_test")
+
def staleness_test(name, outs, generated_pattern, target_files = None, tags = [], **kwargs):
"""Tests that checked-in file(s) match the contents of generated file(s).
@@ -46,7 +48,7 @@ def staleness_test(name, outs, generated_pattern, target_files = None, tags = []
"sed -i.bak -e 's|INSERT_FILE_LIST_HERE|" + "\\\n ".join(file_list) + "|' $@",
)
- native.py_test(
+ py_test(
name = name,
srcs = [script_name],
data = existing_outs + [generated_pattern % file for file in outs],