Skip to content

Commit 449522a

Browse files
committed
minor fixes and linting
1 parent f5602fa commit 449522a

File tree

5 files changed

+20
-11
lines changed

5 files changed

+20
-11
lines changed

src/pip/_internal/req/constructors.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -71,10 +71,14 @@ def _set_requirement_extras(req: Requirement, new_extras: Set[str]) -> Requireme
7171
flags=re.ASCII,
7272
)
7373
# ireq.req is a valid requirement so the regex should always match
74-
assert match is not None, f"regex match on requirement {req} failed, this should never happen"
74+
assert (
75+
match is not None
76+
), f"regex match on requirement {req} failed, this should never happen"
7577
pre: Optional[str] = match.group(1)
7678
post: Optional[str] = match.group(3)
77-
assert pre is not None and post is not None, f"regex group selection for requirement {req} failed, this should never happen"
79+
assert (
80+
pre is not None and post is not None
81+
), f"regex group selection for requirement {req} failed, this should never happen"
7882
extras: str = "[%s]" % ",".join(sorted(new_extras)) if new_extras else ""
7983
return Requirement(f"{pre}{extras}{post}")
8084

src/pip/_internal/resolution/resolvelib/factory.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -468,7 +468,7 @@ def _make_requirements_from_install_req(
468468
yield from ()
469469
elif not ireq.link:
470470
if ireq.extras and ireq.req is not None and ireq.req.specifier:
471-
yield SpecifierWithoutExtrasRequirement(ireq),
471+
yield SpecifierWithoutExtrasRequirement(ireq)
472472
yield SpecifierRequirement(ireq)
473473
else:
474474
self._fail_if_link_is_unsupported_wheel(ireq.link)

src/pip/_internal/resolution/resolvelib/requirements.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -95,7 +95,8 @@ def is_satisfied_by(self, candidate: Candidate) -> bool:
9595

9696
class SpecifierWithoutExtrasRequirement(SpecifierRequirement):
9797
"""
98-
Requirement backed by an install requirement on a base package. Trims extras from its install requirement if there are any.
98+
Requirement backed by an install requirement on a base package.
99+
Trims extras from its install requirement if there are any.
99100
"""
100101

101102
def __init__(self, ireq: InstallRequirement) -> None:

src/pip/_internal/resolution/resolvelib/resolver.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -104,9 +104,13 @@ def resolve(
104104
raise error from e
105105

106106
req_set = RequirementSet(check_supported_wheels=check_supported_wheels)
107-
# process candidates with extras last to ensure their base equivalent is already in the req_set if appropriate
108-
# Python's sort is stable so using a binary key function keeps relative order within both subsets
109-
for candidate in sorted(result.mapping.values(), key=lambda c: c.name != c.project_name):
107+
# process candidates with extras last to ensure their base equivalent is
108+
# already in the req_set if appropriate.
109+
# Python's sort is stable so using a binary key function keeps relative order
110+
# within both subsets.
111+
for candidate in sorted(
112+
result.mapping.values(), key=lambda c: c.name != c.project_name
113+
):
110114
ireq = candidate.get_install_requirement()
111115
if ireq is None:
112116
if candidate.name != candidate.project_name:

tests/unit/resolution_resolvelib/test_requirement.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ def test_new_resolver_requirement_has_name(
6161
) -> None:
6262
"""All requirements should have a name"""
6363
for spec, name, _ in test_cases:
64-
reqs = factory.make_requirements_from_spec(spec, comes_from=None)
64+
reqs = list(factory.make_requirements_from_spec(spec, comes_from=None))
6565
assert len(reqs) == 1
6666
assert reqs[0].name == name
6767

@@ -71,7 +71,7 @@ def test_new_resolver_correct_number_of_matches(
7171
) -> None:
7272
"""Requirements should return the correct number of candidates"""
7373
for spec, _, match_count in test_cases:
74-
reqs = factory.make_requirements_from_spec(spec, comes_from=None)
74+
reqs = list(factory.make_requirements_from_spec(spec, comes_from=None))
7575
assert len(reqs) == 1
7676
req = reqs[0]
7777
matches = factory.find_candidates(
@@ -89,7 +89,7 @@ def test_new_resolver_candidates_match_requirement(
8989
) -> None:
9090
"""Candidates returned from find_candidates should satisfy the requirement"""
9191
for spec, _, _ in test_cases:
92-
reqs = factory.make_requirements_from_spec(spec, comes_from=None)
92+
reqs = list(factory.make_requirements_from_spec(spec, comes_from=None))
9393
assert len(reqs) == 1
9494
req = reqs[0]
9595
candidates = factory.find_candidates(
@@ -106,7 +106,7 @@ def test_new_resolver_candidates_match_requirement(
106106

107107
def test_new_resolver_full_resolve(factory: Factory, provider: PipProvider) -> None:
108108
"""A very basic full resolve"""
109-
reqs = factory.make_requirements_from_spec("simplewheel", comes_from=None)
109+
reqs = list(factory.make_requirements_from_spec("simplewheel", comes_from=None))
110110
assert len(reqs) == 1
111111
r: Resolver[Requirement, Candidate, str] = Resolver(provider, BaseReporter())
112112
result = r.resolve([reqs[0]])

0 commit comments

Comments
 (0)