diff options
Diffstat (limited to 'python/py_test.bzl')
-rw-r--r-- | python/py_test.bzl | 13 |
1 files changed, 12 insertions, 1 deletions
diff --git a/python/py_test.bzl b/python/py_test.bzl index 09580c0..f58f067 100644 --- a/python/py_test.bzl +++ b/python/py_test.bzl @@ -14,7 +14,13 @@ """Public entry point for py_test.""" +load("@rules_python_internal//:rules_python_config.bzl", "config") +load("//python/private:register_extension_info.bzl", "register_extension_info") load("//python/private:util.bzl", "add_migration_tag") +load("//python/private/common:py_test_macro_bazel.bzl", _starlark_py_test = "py_test") + +# buildifier: disable=native-python +_py_test_impl = _starlark_py_test if config.enable_pystar else native.py_test def py_test(**attrs): """See the Bazel core [py_test](https://docs.bazel.build/versions/master/be/python.html#py_test) documentation. @@ -28,4 +34,9 @@ def py_test(**attrs): fail("Python 2 is no longer supported: https://github.com/bazelbuild/rules_python/issues/886") # buildifier: disable=native-python - native.py_test(**add_migration_tag(attrs)) + _py_test_impl(**add_migration_tag(attrs)) + +register_extension_info( + extension = py_test, + label_regex_for_dep = "{extension_name}", +) |