From 54e21a46b6f6c4a0cb234112394b7fe591304c38 Mon Sep 17 00:00:00 2001 From: kyleknap Date: Mon, 29 Jan 2024 22:48:45 -0800 Subject: [PATCH] Look at all sitepackages --- tests/dependencies/test_closure.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/tests/dependencies/test_closure.py b/tests/dependencies/test_closure.py index c7d1b32c816f..a7b831c9631d 100644 --- a/tests/dependencies/test_closure.py +++ b/tests/dependencies/test_closure.py @@ -22,7 +22,6 @@ from packaging.requirements import Requirement _NESTED_STR_DICT = Dict[str, "_NESTED_STR_DICT"] -_SITE_PACKAGES = site.getsitepackages()[0] @pytest.fixture() @@ -82,13 +81,14 @@ def _get_distribution(self, name: str) -> importlib.metadata.Distribution: # Packages names may have a "-". These get converted to "_" for # their respective directory names in the site packages directory. snake_case_name = name.replace("-", "_") - print(_SITE_PACKAGES) - for filename in os.listdir(_SITE_PACKAGES): - print(filename) - if fnmatch.fnmatch(filename, f"{snake_case_name}-*.dist-info"): - return importlib.metadata.Distribution.at( - os.path.join(_SITE_PACKAGES, filename) - ) + for sitepackages in site.getsitepackages(): + print(sitepackages) + for filename in os.listdir(sitepackages): + print(filename) + if fnmatch.fnmatch(filename, f"{snake_case_name}-*.dist-info"): + return importlib.metadata.Distribution.at( + os.path.join(sitepackages, filename) + ) raise ValueError( f'Could not find .dist-info directory for {self.name}' )