Add test cases for manual stub files

Author: brandonchinn178

examples/demo/manual_stubs/explicit/BUILD.bazel

@@ -0,0 +1,13 @@
+load("@rules_python//python:py_library.bzl", "py_library")
+
+py_library(
+    name = "lib",
+    srcs = glob(["lib/**/*.py"]),
+    deps = [":foo-stubs"],
+)
+
+py_library(
+    name = "foo-stubs",
+    imports = ["."],
+    pyi_srcs = glob(["foo/**/*.pyi"]),
+)

examples/demo/manual_stubs/explicit/foo/__init__.pyi

@@ -0,0 +1,2 @@
+class Foo:
+    pass

examples/demo/manual_stubs/explicit/lib/main.py

@@ -0,0 +1 @@
+from foo import Foo

examples/demo/manual_stubs/implicit/BUILD.bazel

@@ -0,0 +1,19 @@
+load("@pip//:requirements.bzl", "requirement")
+load("@rules_python//python:py_library.bzl", "py_library")
+
+py_library(
+    name = "lib",
+    srcs = glob(["lib/**/*.py"]),
+    deps = [
+        # actual dependency here doesn't matter, we just need some dependency
+        # to implicitly provide stubs for; see `py.bzl` for the hook-up
+        requirement("six"),
+    ],
+)
+
+py_library(
+    name = "foo-stubs",
+    imports = ["."],
+    pyi_srcs = glob(["foo/**/*.pyi"]),
+    visibility = ["//visibility:public"],
+)

examples/demo/manual_stubs/implicit/foo/__init__.pyi

@@ -0,0 +1,2 @@
+class Foo:
+    pass

examples/demo/manual_stubs/implicit/lib/main.py

@@ -0,0 +1 @@
+from foo import Foo

examples/demo/py.bzl

@@ -1,8 +1,14 @@
 "Custom py_library rule that also runs mypy."
 
-load("@pip//:requirements.bzl", "all_requirements")
+load("@pip//:requirements.bzl", "all_requirements", "requirement")
 load("@rules_mypy//mypy:mypy.bzl", "load_stubs", "mypy")
 
-stubs = load_stubs(requirements = all_requirements)
+stubs = load_stubs(
+    requirements = all_requirements,
+    overrides = {
+        # See manual_stubs/implicit/
+        requirement("six"): "@@//manual_stubs/implicit:foo-stubs",
+    },
+)
 
 mypy_aspect = mypy(stubs = stubs)

examples/demo/requirements.in

@@ -1,3 +1,7 @@
-cachetools~=5.4.0
+# ----- thm/ dependencies ----- #
 numpy~=1.26.4
+cachetools~=5.4.0
 types-cachetools~=5.4.0.20240717
+
+# ---- manual_stubs/ dependencies ----- #
+six==1.17.0

examples/demo/requirements.txt

@@ -44,6 +44,10 @@ numpy==1.26.4 \
     --hash=sha256:f870204a840a60da0b12273ef34f7051e98c3b5961b61b0c2c1be6dfd64fbcd3 \
     --hash=sha256:ffa75af20b44f8dba823498024771d5ac50620e6915abac414251bd971b4529f
     # via -r requirements.in
+six==1.17.0 \
+    --hash=sha256:4721f391ed90541fddacab5acf947aa0d3dc7d27b2e1e8eda2be8970586c3274 \
+    --hash=sha256:ff70335d468e7eb6ec65b95b99d3a2836546063f63acc5171de367e834932a81
+    # via -r requirements.in
 types-cachetools==5.4.0.20240717 \
     --hash=sha256:1eae90c48760bac44ab89108be938e8ce1d740910f2d4b68446dcdc82763f186 \
     --hash=sha256:67c84c26df988039be68344b162afd2dd7cd3741dc08e7d67aa1954782fd2d2a