Skip to content

Commit 9262b33

Browse files
mkniewallnerfpgmaas
andcommitted
refactor(distribution): better name methods
Co-authored-by: Florian Maas <[email protected]>
1 parent 981bd18 commit 9262b33

File tree

2 files changed

+22
-20
lines changed

2 files changed

+22
-20
lines changed

python/deptry/distribution.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ def normalize_distribution_name(name: str) -> str:
1616

1717

1818
@lru_cache(maxsize=1)
19-
def get_packages_normalized_distributions() -> dict[str, set[str]]:
19+
def get_packages_to_normalized_distributions_mapping() -> dict[str, set[str]]:
2020
"""
2121
Return a mapping of top-level packages to their normalized distributions.
2222
Cache ensures that we only build this mapping once, since it should not change during the invocation of deptry.
@@ -28,14 +28,14 @@ def get_packages_normalized_distributions() -> dict[str, set[str]]:
2828

2929

3030
@lru_cache(maxsize=1)
31-
def get_normalized_distributions_packages() -> dict[str, set[str]]:
31+
def get_normalized_distributions_to_packages_mapping() -> dict[str, set[str]]:
3232
"""
3333
Return a mapping of normalized distributions to their top-level packages.
3434
Cache ensures that we only build this mapping once, since it should not change during the invocation of deptry.
3535
"""
3636
distributions_packages: dict[str, set[str]] = defaultdict(set)
3737

38-
for package, distributions in get_packages_normalized_distributions().items():
38+
for package, distributions in get_packages_to_normalized_distributions_mapping().items():
3939
for distribution in distributions:
4040
distributions_packages[distribution].add(package)
4141

@@ -46,11 +46,11 @@ def get_distributions_from_package(name: str) -> set[str] | None:
4646
"""
4747
Retrieve the distributions provided by the package, if any.
4848
"""
49-
return get_packages_normalized_distributions().get(name)
49+
return get_packages_to_normalized_distributions_mapping().get(name)
5050

5151

5252
def get_packages_from_distribution(name: str) -> set[str] | None:
5353
"""
5454
Normalize the distribution and retrieve the packages it provides, if any.
5555
"""
56-
return get_normalized_distributions_packages().get(normalize_distribution_name(name))
56+
return get_normalized_distributions_to_packages_mapping().get(normalize_distribution_name(name))

tests/unit/test_distribution.py

Lines changed: 17 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -6,9 +6,9 @@
66

77
from deptry.distribution import (
88
get_distributions_from_package,
9-
get_normalized_distributions_packages,
9+
get_normalized_distributions_to_packages_mapping,
1010
get_packages_from_distribution,
11-
get_packages_normalized_distributions,
11+
get_packages_to_normalized_distributions_mapping,
1212
normalize_distribution_name,
1313
)
1414

@@ -29,9 +29,9 @@ def test_normalize_distribution_name(name: str) -> None:
2929
assert normalize_distribution_name(name) == "friendly-bard"
3030

3131

32-
def test_get_packages_normalized_distributions() -> None:
32+
def test_get_packages_to_normalized_distributions_mapping() -> None:
3333
# Clear cache before calling the function, as it is also populated during testing.
34-
get_packages_normalized_distributions.cache_clear()
34+
get_packages_to_normalized_distributions_mapping.cache_clear()
3535

3636
with patch(
3737
"deptry.distribution.importlib_metadata.packages_distributions",
@@ -44,13 +44,13 @@ def test_get_packages_normalized_distributions() -> None:
4444
"setuptools": ["setuptools"],
4545
},
4646
) as mock_packages_distributions:
47-
normalized_packages_distributions = get_packages_normalized_distributions()
47+
normalized_packages_distributions = get_packages_to_normalized_distributions_mapping()
4848

4949
# Call function a second time, to ensure that we only call `packages_distributions` once.
50-
get_packages_normalized_distributions()
50+
get_packages_to_normalized_distributions_mapping()
5151

5252
# Clear cache after calling the function to avoid keeping our mocked values, in case test invocation depend on it.
53-
get_packages_normalized_distributions.cache_clear()
53+
get_packages_to_normalized_distributions_mapping.cache_clear()
5454

5555
assert normalized_packages_distributions == {
5656
"requests": {"requests"},
@@ -63,12 +63,12 @@ def test_get_packages_normalized_distributions() -> None:
6363
mock_packages_distributions.assert_called_once()
6464

6565

66-
def test_get_normalized_distributions_packages() -> None:
66+
def test_get_normalized_distributions_to_packages_mapping() -> None:
6767
# Clear cache before calling the function, as it is also populated during testing.
68-
get_normalized_distributions_packages.cache_clear()
68+
get_normalized_distributions_to_packages_mapping.cache_clear()
6969

7070
with patch(
71-
"deptry.distribution.get_packages_normalized_distributions",
71+
"deptry.distribution.get_packages_to_normalized_distributions_mapping",
7272
return_value={
7373
"requests": {"requests"},
7474
"charset_normalizer": {"charset-normalizer"},
@@ -78,13 +78,13 @@ def test_get_normalized_distributions_packages() -> None:
7878
"setuptools": {"setuptools"},
7979
},
8080
) as mock_packages_distributions:
81-
distributions_packages = get_normalized_distributions_packages()
81+
distributions_packages = get_normalized_distributions_to_packages_mapping()
8282

8383
# Call function a second time, to ensure that we only call `packages_distributions` once.
84-
get_normalized_distributions_packages()
84+
get_normalized_distributions_to_packages_mapping()
8585

8686
# Clear cache after calling the function to avoid keeping our mocked values, in case test invocation depend on it.
87-
get_normalized_distributions_packages.cache_clear()
87+
get_normalized_distributions_to_packages_mapping.cache_clear()
8888

8989
assert distributions_packages == {
9090
"requests": {"requests"},
@@ -97,7 +97,7 @@ def test_get_normalized_distributions_packages() -> None:
9797

9898
def test_get_distributions_from_package() -> None:
9999
with patch(
100-
"deptry.distribution.get_packages_normalized_distributions",
100+
"deptry.distribution.get_packages_to_normalized_distributions_mapping",
101101
return_value={
102102
"bar": {"foo-bar"},
103103
"foo": {"foo-bar", "foo"},
@@ -107,5 +107,7 @@ def test_get_distributions_from_package() -> None:
107107

108108

109109
def test_get_packages_from_distribution() -> None:
110-
with patch("deptry.distribution.get_normalized_distributions_packages", return_value={"foo-bar": {"bar", "foo"}}):
110+
with patch(
111+
"deptry.distribution.get_normalized_distributions_to_packages_mapping", return_value={"foo-bar": {"bar", "foo"}}
112+
):
111113
assert get_packages_from_distribution("foo_Bar") == {"bar", "foo"}

0 commit comments

Comments
 (0)