more tests

python · Oct 16, 2024 · 6ea5816 · 6ea5816
1 parent a47c7db
commit 6ea5816
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/mypy/test/teststubinfo.py b/mypy/test/teststubinfo.py
@@ -24,6 +24,9 @@ def test_approved_stub_package_exists(self) -> None:
         assert approved_stub_package_exists("google.cloud.ndb")
         assert approved_stub_package_exists("google.cloud.ndb.submodule")
         assert not approved_stub_package_exists("google.cloud.unknown")
+        assert approved_stub_package_exists("google.protobuf")
+        assert approved_stub_package_exists("google.protobuf.submodule")
+        assert not approved_stub_package_exists("google")
 
     def test_stub_distribution_name(self) -> None:
         assert stub_distribution_name("foobar_asdf") is None
@@ -32,6 +35,9 @@ def test_stub_distribution_name(self) -> None:
         assert stub_distribution_name("google.cloud.ndb") == "types-google-cloud-ndb"
         assert stub_distribution_name("google.cloud.ndb.submodule") == "types-google-cloud-ndb"
         assert stub_distribution_name("google.cloud.unknown") is None
+        assert stub_distribution_name("google.protobuf") == "types-protobuf"
+        assert stub_distribution_name("google.protobuf.submodule") == "types-protobuf"
+        assert stub_distribution_name("google") is None
 
     def test_period_in_top_level(self) -> None:
         for packages in (non_bundled_packages_flat, legacy_bundled_packages):