tighten typechecking (#6203)

2deaf594 · David Hotham · GitHub · 3bd4c7a3 · 2deaf594 · 2deaf594
Commit 2deaf594 authored Aug 20, 2022 by David Hotham Committed by GitHub Aug 21, 2022
14 changed files
--- a/src/poetry/puzzle/provider.py
+++ b/src/poetry/puzzle/provider.py
@@ -410,7 +410,7 @@ class Provider:
        file_name = os.path.basename(urllib.parse.urlparse(url).path)
        with tempfile.TemporaryDirectory() as temp_dir:
            dest = Path(temp_dir) / file_name
-            download_file(url, str(dest))
+            download_file(url, dest)
            package = cls.get_package_from_file(dest)
        package._source_type = "url"
@@ -528,7 +528,7 @@ class Provider:
                    dependency,
                    self._pool.package(
                        package.name,
-                        package.version.text,
+                        package.version,
                        extras=list(dependency.extras),
                        repository=dependency.source_name,
                    ),

--- a/src/poetry/repositories/cached.py
+++ b/src/poetry/repositories/cached.py
@@ -13,7 +13,9 @@ from poetry.repositories.repository import Repository
 if TYPE_CHECKING:
+    from packaging.utils import NormalizedName
    from poetry.core.packages.package import Package
+    from poetry.core.semver.version import Version
    from poetry.inspection.info import PackageInfo
@@ -40,10 +42,12 @@ class CachedRepository(Repository, ABC):
        )
    @abstractmethod
-    def _get_release_info(self, name: str, version: str) -> dict[str, Any]:
+    def _get_release_info(
+        self, name: NormalizedName, version: Version
+    ) -> dict[str, Any]:
        raise NotImplementedError()
-    def get_release_info(self, name: str, version: str) -> PackageInfo:
+    def get_release_info(self, name: NormalizedName, version: Version) -> PackageInfo:
        """
        Return the release information given a package name and a version.
@@ -74,8 +78,8 @@ class CachedRepository(Repository, ABC):
    def package(
        self,
-        name: str,
+        name: NormalizedName,
-        version: str,
+        version: Version,
        extras: list[str] | None = None,
    ) -> Package:
        return self.get_release_info(name, version).to_package(name=name, extras=extras)
--- a/src/poetry/repositories/http.py
+++ b/src/poetry/repositories/http.py
@@ -68,7 +68,7 @@ class HTTPRepository(CachedRepository, ABC):
    def authenticated_url(self) -> str:
        return self._authenticator.authenticated_url(url=self.url)
-    def _download(self, url: str, dest: str) -> None:
+    def _download(self, url: str, dest: Path) -> None:
        return download_file(url, dest, session=self.session)
    def _get_info_from_wheel(self, url: str) -> PackageInfo:
@@ -81,7 +81,7 @@ class HTTPRepository(CachedRepository, ABC):
        with temporary_directory() as temp_dir:
            filepath = Path(temp_dir) / filename
-            self._download(url, str(filepath))
+            self._download(url, filepath)
            return PackageInfo.from_wheel(filepath)
@@ -97,7 +97,7 @@ class HTTPRepository(CachedRepository, ABC):
        with temporary_directory() as temp_dir:
            filepath = Path(temp_dir) / filename
-            self._download(url, str(filepath))
+            self._download(url, filepath)
            return PackageInfo.from_sdist(filepath)
@@ -226,7 +226,7 @@ class HTTPRepository(CachedRepository, ABC):
            ):
                with temporary_directory() as temp_dir:
                    filepath = Path(temp_dir) / link.filename
-                    self._download(link.url, str(filepath))
+                    self._download(link.url, filepath)
                    known_hash = (
                        getattr(hashlib, link.hash_name)() if link.hash_name else None

--- a/src/poetry/repositories/legacy_repository.py
+++ b/src/poetry/repositories/legacy_repository.py
@@ -3,9 +3,7 @@ from __future__ import annotations
 from typing import TYPE_CHECKING
 from typing import Any
-from packaging.utils import canonicalize_name
 from poetry.core.packages.package import Package
-from poetry.core.semver.version import Version
 from poetry.inspection.info import PackageInfo
 from poetry.repositories.exceptions import PackageNotFound
@@ -16,6 +14,7 @@ from poetry.repositories.link_sources.html import SimpleRepositoryPage
 if TYPE_CHECKING:
    from packaging.utils import NormalizedName
    from poetry.core.packages.utils.link import Link
+    from poetry.core.semver.version import Version
    from poetry.core.semver.version_constraint import VersionConstraint
    from poetry.config.config import Config
@@ -35,7 +34,7 @@ class LegacyRepository(HTTPRepository):
        super().__init__(name, url.rstrip("/"), config, disable_cache)
    def package(
-        self, name: str, version: str, extras: list[str] | None = None
+        self, name: NormalizedName, version: Version, extras: list[str] | None = None
    ) -> Package:
        """
        Retrieve the release information.
@@ -49,7 +48,7 @@ class LegacyRepository(HTTPRepository):
        should be much faster.
        """
        try:
-            index = self._packages.index(Package(name, version, version))
+            index = self._packages.index(Package(name, version))
            return self._packages[index]
        except ValueError:
@@ -106,18 +105,20 @@ class LegacyRepository(HTTPRepository):
            for version in versions
        ]
-    def _get_release_info(self, name: str, version: str) -> dict[str, Any]:
+    def _get_release_info(
-        page = self._get_page(f"/{canonicalize_name(name)}/")
+        self, name: NormalizedName, version: Version
+    ) -> dict[str, Any]:
+        page = self._get_page(f"/{name}/")
        if page is None:
            raise PackageNotFound(f'No package named "{name}"')
-        links = list(page.links_for_version(name, Version.parse(version)))
+        links = list(page.links_for_version(name, version))
        return self._links_to_data(
            links,
            PackageInfo(
                name=name,
-                version=version,
+                version=version.text,
                summary="",
                platform=None,
                requires_dist=[],

--- a/src/poetry/repositories/link_sources/base.py
+++ b/src/poetry/repositories/link_sources/base.py
@@ -17,6 +17,7 @@ from poetry.utils.patterns import wheel_file_re
 if TYPE_CHECKING:
    from collections.abc import Iterator
+    from packaging.utils import NormalizedName
    from poetry.core.packages.utils.link import Link
@@ -98,9 +99,9 @@ class LinkSource:
            pkg = Package(name, version, source_url=link.url)
        return pkg
-    def links_for_version(self, name: str, version: Version) -> Iterator[Link]:
+    def links_for_version(
-        name = canonicalize_name(name)
+        self, name: NormalizedName, version: Version
+    ) -> Iterator[Link]:
        for link in self.links:
            pkg = self.link_package_data(link)

--- a/src/poetry/repositories/pool.py
+++ b/src/poetry/repositories/pool.py
@@ -7,8 +7,10 @@ from poetry.repositories.repository import Repository
 if TYPE_CHECKING:
+    from packaging.utils import NormalizedName
    from poetry.core.packages.dependency import Dependency
    from poetry.core.packages.package import Package
+    from poetry.core.semver.version import Version
 class Pool(Repository):
@@ -118,8 +120,8 @@ class Pool(Repository):
    def package(
        self,
-        name: str,
+        name: NormalizedName,
-        version: str,
+        version: Version,
        extras: list[str] | None = None,
        repository: str | None = None,
    ) -> Package:

--- a/src/poetry/repositories/pypi_repository.py
+++ b/src/poetry/repositories/pypi_repository.py
@@ -168,7 +168,7 @@ class PyPiRepository(HTTPRepository):
        return links
    def _get_release_info(
-        self, name: str, version: str
+        self, name: NormalizedName, version: Version
    ) -> dict[str, str | list[str] | None]:
        from poetry.inspection.info import PackageInfo

--- a/src/poetry/repositories/repository.py
+++ b/src/poetry/repositories/repository.py
@@ -16,6 +16,7 @@ if TYPE_CHECKING:
    from poetry.core.packages.dependency import Dependency
    from poetry.core.packages.package import Package
    from poetry.core.packages.utils.link import Link
+    from poetry.core.semver.version import Version
 class Repository:
@@ -133,12 +134,10 @@ class Repository:
        return []
    def package(
-        self, name: str, version: str, extras: list[str] | None = None
+        self, name: NormalizedName, version: Version, extras: list[str] | None = None
    ) -> Package:
-        name = name.lower()
        for package in self.packages:
-            if name == package.name and package.version.text == version:
+            if name == package.name and package.version == version:
                return package.clone()
        raise PackageNotFound(f"Package {name} ({version}) not found.")
--- a/src/poetry/utils/helpers.py
+++ b/src/poetry/utils/helpers.py
@@ -72,7 +72,7 @@ def merge_dicts(d1: dict[str, Any], d2: dict[str, Any]) -> None:
 def download_file(
    url: str,
-    dest: str,
+    dest: Path,
    session: Authenticator | Session | None = None,
    chunk_size: int = 1024,
 ) -> None:

--- a/tests/helpers.py
+++ b/tests/helpers.py
@@ -125,13 +125,13 @@ def mock_clone(
    return MockDulwichRepo(dest)
-def mock_download(url: str, dest: str, **__: Any) -> None:
+def mock_download(url: str, dest: Path) -> None:
    parts = urllib.parse.urlparse(url)
    fixtures = Path(__file__).parent / "fixtures"
    fixture = fixtures / parts.path.lstrip("/")
-    copy_or_symlink(fixture, Path(dest))
+    copy_or_symlink(fixture, dest)
 class TestExecutor(Executor):

--- a/tests/masonry/builders/test_editable_builder.py
+++ b/tests/masonry/builders/test_editable_builder.py
@@ -12,6 +12,8 @@ import pytest
 from cleo.io.null_io import NullIO
 from deepdiff import DeepDiff
+from packaging.utils import canonicalize_name
+from poetry.core.semver.version import Version
 from poetry.factory import Factory
 from poetry.masonry.builders.editable import EditableBuilder
@@ -224,7 +226,7 @@ def test_builder_falls_back_on_setup_and_pip_for_packages_with_build_scripts(
    assert [] == env.executed
-def test_builder_setup_generation_runs_with_pip_editable(tmp_dir: str):
+def test_builder_setup_generation_runs_with_pip_editable(tmp_dir: str) -> None:
    # create an isolated copy of the project
    fixture = Path(__file__).parent.parent.parent / "fixtures" / "extended_project"
    extended_project = Path(tmp_dir) / "extended_project"
@@ -241,7 +243,10 @@ def test_builder_setup_generation_runs_with_pip_editable(tmp_dir: str):
        # is the package installed?
        repository = InstalledRepository.load(venv)
-        assert repository.package("extended-project", "1.2.3")
+        package = repository.package(
+            canonicalize_name("extended-project"), Version.parse("1.2.3")
+        )
+        assert package.name == "extended-project"
        # check for the module built by build.py
        try:

--- a/tests/repositories/test_legacy_repository.py
+++ b/tests/repositories/test_legacy_repository.py
@@ -10,7 +10,9 @@ from typing import TYPE_CHECKING
 import pytest
 import requests
+from packaging.utils import canonicalize_name
 from poetry.core.packages.dependency import Dependency
+from poetry.core.semver.version import Version
 from poetry.factory import Factory
 from poetry.repositories.exceptions import PackageNotFound
@@ -49,7 +51,7 @@ class MockRepository(LegacyRepository):
        fixture = self.FIXTURES / (name + ".html")
        if not fixture.exists():
-            return
+            return None
        with fixture.open(encoding="utf-8") as f:
            return SimpleRepositoryPage(self._url + endpoint, f.read())
@@ -61,39 +63,43 @@ class MockRepository(LegacyRepository):
        shutil.copyfile(str(filepath), dest)
-def test_page_relative_links_path_are_correct():
+def test_page_relative_links_path_are_correct() -> None:
    repo = MockRepository()
    page = repo._get_page("/relative")
+    assert page is not None
    for link in page.links:
        assert link.netloc == "legacy.foo.bar"
        assert link.path.startswith("/relative/poetry")
-def test_page_absolute_links_path_are_correct():
+def test_page_absolute_links_path_are_correct() -> None:
    repo = MockRepository()
    page = repo._get_page("/absolute")
+    assert page is not None
    for link in page.links:
        assert link.netloc == "files.pythonhosted.org"
        assert link.path.startswith("/packages/")
-def test_page_clean_link():
+def test_page_clean_link() -> None:
    repo = MockRepository()
    page = repo._get_page("/relative")
+    assert page is not None
    cleaned = page.clean_link('https://legacy.foo.bar/test /the"/cleaning\0')
    assert cleaned == "https://legacy.foo.bar/test%20/the%22/cleaning%00"
-def test_page_invalid_version_link():
+def test_page_invalid_version_link() -> None:
    repo = MockRepository()
    page = repo._get_page("/invalid-version")
+    assert page is not None
    links = list(page.links)
    assert len(links) == 2
@@ -109,7 +115,7 @@ def test_page_invalid_version_link():
            invalid_link = link
            break
-    links_010 = list(page.links_for_version("poetry", versions[0]))
+    links_010 = list(page.links_for_version(canonicalize_name("poetry"), versions[0]))
    assert invalid_link not in links_010
    assert invalid_link
@@ -121,25 +127,28 @@ def test_page_invalid_version_link():
    assert packages[0].version.to_string() == "0.1.0"
-def test_sdist_format_support():
+def test_sdist_format_support() -> None:
    repo = MockRepository()
    page = repo._get_page("/relative")
+    assert page is not None
    bz2_links = list(filter(lambda link: link.ext == ".tar.bz2", page.links))
    assert len(bz2_links) == 1
    assert bz2_links[0].filename == "poetry-0.1.1.tar.bz2"
-def test_missing_version():
+def test_missing_version() -> None:
    repo = MockRepository()
    with pytest.raises(PackageNotFound):
-        repo._get_release_info("missing_version", "1.1.0")
+        repo._get_release_info(
+            canonicalize_name("missing_version"), Version.parse("1.1.0")
+        )
-def test_get_package_information_fallback_read_setup():
+def test_get_package_information_fallback_read_setup() -> None:
    repo = MockRepository()
-    package = repo.package("jupyter", "1.0.0")
+    package = repo.package(canonicalize_name("jupyter"), Version.parse("1.0.0"))
    assert package.source_type == "legacy"
    assert package.source_reference == repo.name
@@ -152,10 +161,12 @@ def test_get_package_information_fallback_read_setup():
    )
-def test_get_package_information_skips_dependencies_with_invalid_constraints():
+def test_get_package_information_skips_dependencies_with_invalid_constraints() -> None:
    repo = MockRepository()
-    package = repo.package("python-language-server", "0.21.2")
+    package = repo.package(
+        canonicalize_name("python-language-server"), Version.parse("0.21.2")
+    )
    assert package.name == "python-language-server"
    assert package.version.text == "0.21.2"
@@ -188,7 +199,7 @@ def test_get_package_information_skips_dependencies_with_invalid_constraints():
    ]
-def test_find_packages_no_prereleases():
+def test_find_packages_no_prereleases() -> None:
    repo = MockRepository()
    packages = repo.find_packages(Factory.create_dependency("pyyaml", "*"))
@@ -203,7 +214,7 @@ def test_find_packages_no_prereleases():
 @pytest.mark.parametrize(
    ["constraint", "count"], [("*", 1), (">=1", 0), (">=19.0.0a0", 1)]
 )
-def test_find_packages_only_prereleases(constraint: str, count: int):
+def test_find_packages_only_prereleases(constraint: str, count: int) -> None:
    repo = MockRepository()
    packages = repo.find_packages(Factory.create_dependency("black", constraint))
@@ -216,17 +227,17 @@ def test_find_packages_only_prereleases(constraint: str, count: int):
            assert package.source_url == repo.url
-def test_find_packages_only_prereleases_empty_when_not_any():
+def test_find_packages_only_prereleases_empty_when_not_any() -> None:
    repo = MockRepository()
    packages = repo.find_packages(Factory.create_dependency("black", ">=1"))
    assert len(packages) == 0
-def test_get_package_information_chooses_correct_distribution():
+def test_get_package_information_chooses_correct_distribution() -> None:
    repo = MockRepository()
-    package = repo.package("isort", "4.3.4")
+    package = repo.package(canonicalize_name("isort"), Version.parse("4.3.4"))
    assert package.name == "isort"
    assert package.version.text == "4.3.4"
@@ -236,30 +247,32 @@ def test_get_package_information_chooses_correct_distribution():
    assert futures_dep.python_versions == "~2.7"
-def test_get_package_information_includes_python_requires():
+def test_get_package_information_includes_python_requires() -> None:
    repo = MockRepository()
-    package = repo.package("futures", "3.2.0")
+    package = repo.package(canonicalize_name("futures"), Version.parse("3.2.0"))
    assert package.name == "futures"
    assert package.version.text == "3.2.0"
    assert package.python_versions == ">=2.6, <3"
-def test_get_package_information_sets_appropriate_python_versions_if_wheels_only():
+def test_get_package_information_sets_appropriate_python_versions_if_wheels_only() -> (
+    None
+):
    repo = MockRepository()
-    package = repo.package("futures", "3.2.0")
+    package = repo.package(canonicalize_name("futures"), Version.parse("3.2.0"))
    assert package.name == "futures"
    assert package.version.text == "3.2.0"
    assert package.python_versions == ">=2.6, <3"
-def test_get_package_from_both_py2_and_py3_specific_wheels():
+def test_get_package_from_both_py2_and_py3_specific_wheels() -> None:
    repo = MockRepository()
-    package = repo.package("ipython", "5.7.0")
+    package = repo.package(canonicalize_name("ipython"), Version.parse("5.7.0"))
    assert package.name == "ipython"
    assert package.version.text == "5.7.0"
@@ -294,20 +307,22 @@ def test_get_package_from_both_py2_and_py3_specific_wheels():
    assert str(required[5].marker) == 'sys_platform != "win32"'
-def test_get_package_from_both_py2_and_py3_specific_wheels_python_constraint():
+def test_get_package_from_both_py2_and_py3_specific_wheels_python_constraint() -> None:
    repo = MockRepository()
-    package = repo.package("poetry-test-py2-py3-metadata-merge", "0.1.0")
+    package = repo.package(
+        canonicalize_name("poetry-test-py2-py3-metadata-merge"), Version.parse("0.1.0")
+    )
    assert package.name == "poetry-test-py2-py3-metadata-merge"
    assert package.version.text == "0.1.0"
    assert package.python_versions == ">=2.7,<2.8 || >=3.7,<4.0"
-def test_get_package_with_dist_and_universal_py3_wheel():
+def test_get_package_with_dist_and_universal_py3_wheel() -> None:
    repo = MockRepository()
-    package = repo.package("ipython", "7.5.0")
+    package = repo.package(canonicalize_name("ipython"), Version.parse("7.5.0"))
    assert package.name == "ipython"
    assert package.version.text == "7.5.0"
@@ -332,10 +347,10 @@ def test_get_package_with_dist_and_universal_py3_wheel():
    assert sorted(required, key=lambda dep: dep.name) == expected
-def test_get_package_retrieves_non_sha256_hashes():
+def test_get_package_retrieves_non_sha256_hashes() -> None:
    repo = MockRepository()
-    package = repo.package("ipython", "7.5.0")
+    package = repo.package(canonicalize_name("ipython"), Version.parse("7.5.0"))
    expected = [
        {
@@ -351,10 +366,10 @@ def test_get_package_retrieves_non_sha256_hashes():
    assert package.files == expected
-def test_get_package_retrieves_non_sha256_hashes_mismatching_known_hash():
+def test_get_package_retrieves_non_sha256_hashes_mismatching_known_hash() -> None:
    repo = MockRepository()
-    package = repo.package("ipython", "5.7.0")
+    package = repo.package(canonicalize_name("ipython"), Version.parse("5.7.0"))
    expected = [
        {
@@ -374,10 +389,10 @@ def test_get_package_retrieves_non_sha256_hashes_mismatching_known_hash():
    assert package.files == expected
-def test_get_package_retrieves_packages_with_no_hashes():
+def test_get_package_retrieves_packages_with_no_hashes() -> None:
    repo = MockRepository()
-    package = repo.package("jupyter", "1.0.0")
+    package = repo.package(canonicalize_name("jupyter"), Version.parse("1.0.0"))
    assert [
        {
@@ -399,20 +414,22 @@ class MockHttpRepository(LegacyRepository):
            http.register_uri(http.GET, url, status=response)
-def test_get_200_returns_page(http: type[httpretty.httpretty]):
+def test_get_200_returns_page(http: type[httpretty.httpretty]) -> None:
    repo = MockHttpRepository({"/foo": 200}, http)
    assert repo._get_page("/foo")
 @pytest.mark.parametrize("status_code", [401, 403, 404])
-def test_get_40x_and_returns_none(http: type[httpretty.httpretty], status_code: int):
+def test_get_40x_and_returns_none(
+    http: type[httpretty.httpretty], status_code: int
+) -> None:
    repo = MockHttpRepository({"/foo": status_code}, http)
    assert repo._get_page("/foo") is None
-def test_get_5xx_raises(http: type[httpretty.httpretty]):
+def test_get_5xx_raises(http: type[httpretty.httpretty]) -> None:
    repo = MockHttpRepository({"/foo": 500}, http)
    with pytest.raises(RepositoryError):
@@ -421,7 +438,7 @@ def test_get_5xx_raises(http: type[httpretty.httpretty]):
 def test_get_redirected_response_url(
    http: type[httpretty.httpretty], monkeypatch: MonkeyPatch
-):
+) -> None:
    repo = MockHttpRepository({"/foo": 200}, http)
    redirect_url = "http://legacy.redirect.bar"
@@ -434,7 +451,9 @@ def test_get_redirected_response_url(
        return response
    monkeypatch.setattr(repo.session, "get", get_mock)
-    assert repo._get_page("/foo")._url == "http://legacy.redirect.bar/foo/"
+    page = repo._get_page("/foo")
+    assert page is not None
+    assert page._url == "http://legacy.redirect.bar/foo/"
 @pytest.mark.parametrize(

--- a/tests/repositories/test_pool.py
+++ b/tests/repositories/test_pool.py
@@ -2,18 +2,21 @@ from __future__ import annotations
 import pytest
+from packaging.utils import canonicalize_name
+from poetry.core.semver.version import Version
 from poetry.repositories import Pool
 from poetry.repositories import Repository
 from poetry.repositories.exceptions import PackageNotFound
 from poetry.repositories.legacy_repository import LegacyRepository
-def test_pool_raises_package_not_found_when_no_package_is_found():
+def test_pool_raises_package_not_found_when_no_package_is_found() -> None:
    pool = Pool()
    pool.add_repository(Repository("repo"))
    with pytest.raises(PackageNotFound):
-        pool.package("foo", "1.0.0")
+        pool.package(canonicalize_name("foo"), Version.parse("1.0.0"))
 def test_pool():

--- a/tests/repositories/test_pypi_repository.py
+++ b/tests/repositories/test_pypi_repository.py
@@ -9,7 +9,9 @@ from typing import TYPE_CHECKING
 import pytest
+from packaging.utils import canonicalize_name
 from poetry.core.packages.dependency import Dependency
+from poetry.core.semver.version import Version
 from requests.exceptions import TooManyRedirects
 from requests.models import Response
@@ -50,7 +52,7 @@ class MockRepository(PyPiRepository):
                fixture = self.JSON_FIXTURES / (name + ".json")
        if not fixture.exists():
-            return
+            return None
        with fixture.open(encoding="utf-8") as f:
            return json.loads(f.read())
@@ -63,21 +65,21 @@ class MockRepository(PyPiRepository):
        shutil.copyfile(str(fixture), dest)
-def test_find_packages():
+def test_find_packages() -> None:
    repo = MockRepository()
    packages = repo.find_packages(Factory.create_dependency("requests", "^2.18"))
    assert len(packages) == 5
-def test_find_packages_with_prereleases():
+def test_find_packages_with_prereleases() -> None:
    repo = MockRepository()
    packages = repo.find_packages(Factory.create_dependency("toga", ">=0.3.0.dev2"))
    assert len(packages) == 7
-def test_find_packages_does_not_select_prereleases_if_not_allowed():
+def test_find_packages_does_not_select_prereleases_if_not_allowed() -> None:
    repo = MockRepository()
    packages = repo.find_packages(Factory.create_dependency("pyyaml", "*"))
@@ -87,17 +89,17 @@ def test_find_packages_does_not_select_prereleases_if_not_allowed():
 @pytest.mark.parametrize(
    ["constraint", "count"], [("*", 1), (">=1", 0), (">=19.0.0a0", 1)]
 )
-def test_find_packages_only_prereleases(constraint: str, count: int):
+def test_find_packages_only_prereleases(constraint: str, count: int) -> None:
    repo = MockRepository()
    packages = repo.find_packages(Factory.create_dependency("black", constraint))
    assert len(packages) == count
-def test_package():
+def test_package() -> None:
    repo = MockRepository()
-    package = repo.package("requests", "2.18.4")
+    package = repo.package(canonicalize_name("requests"), Version.parse("2.18.4"))
    assert package.name == "requests"
    assert len(package.requires) == 9
@@ -126,10 +128,10 @@ def test_package():
    )
-def test_fallback_on_downloading_packages():
+def test_fallback_on_downloading_packages() -> None:
    repo = MockRepository(fallback=True)
-    package = repo.package("jupyter", "1.0.0")
+    package = repo.package(canonicalize_name("jupyter"), Version.parse("1.0.0"))
    assert package.name == "jupyter"
    assert len(package.requires) == 6
@@ -145,10 +147,10 @@ def test_fallback_on_downloading_packages():
    ]
-def test_fallback_inspects_sdist_first_if_no_matching_wheels_can_be_found():
+def test_fallback_inspects_sdist_first_if_no_matching_wheels_can_be_found() -> None:
    repo = MockRepository(fallback=True)
-    package = repo.package("isort", "4.3.4")
+    package = repo.package(canonicalize_name("isort"), Version.parse("4.3.4"))
    assert package.name == "isort"
    assert len(package.requires) == 1
@@ -158,10 +160,10 @@ def test_fallback_inspects_sdist_first_if_no_matching_wheels_can_be_found():
    assert dep.python_versions == "~2.7"
-def test_fallback_can_read_setup_to_get_dependencies():
+def test_fallback_can_read_setup_to_get_dependencies() -> None:
    repo = MockRepository(fallback=True)
-    package = repo.package("sqlalchemy", "1.2.12")
+    package = repo.package(canonicalize_name("sqlalchemy"), Version.parse("1.2.12"))
    assert package.name == "sqlalchemy"
    assert len(package.requires) == 9
@@ -180,10 +182,10 @@ def test_fallback_can_read_setup_to_get_dependencies():
    }
-def test_pypi_repository_supports_reading_bz2_files():
+def test_pypi_repository_supports_reading_bz2_files() -> None:
    repo = MockRepository(fallback=True)
-    package = repo.package("twisted", "18.9.0")
+    package = repo.package(canonicalize_name("twisted"), Version.parse("18.9.0"))
    assert package.name == "twisted"
    assert len(package.requires) == 71
@@ -220,7 +222,7 @@ def test_pypi_repository_supports_reading_bz2_files():
        )
-def test_invalid_versions_ignored():
+def test_invalid_versions_ignored() -> None:
    repo = MockRepository()
    # the json metadata for this package contains one malformed version
@@ -229,7 +231,9 @@ def test_invalid_versions_ignored():
    assert len(packages) == 1
-def test_get_should_invalid_cache_on_too_many_redirects_error(mocker: MockerFixture):
+def test_get_should_invalid_cache_on_too_many_redirects_error(
+    mocker: MockerFixture,
+) -> None:
    delete_cache = mocker.patch("cachecontrol.caches.file_cache.FileCache.delete")
    response = Response()
@@ -246,14 +250,14 @@ def test_get_should_invalid_cache_on_too_many_redirects_error(mocker: MockerFixt
    assert delete_cache.called
-def test_urls():
+def test_urls() -> None:
    repository = PyPiRepository()
    assert repository.url == "https://pypi.org/simple/"
    assert repository.authenticated_url == "https://pypi.org/simple/"
-def test_use_pypi_pretty_name():
+def test_use_pypi_pretty_name() -> None:
    repo = MockRepository(fallback=True)
    package = repo.find_packages(Factory.create_dependency("twisted", "*"))