def lib_dir(name, wheeldir, user=False, home=None, root=None,
isolated=False, prefix=None):
from pip._internal.locations import distutils_scheme
from pip._internal.wheel import root_is_purelib
scheme = distutils_scheme(
"", user=user, home=home, root=root,
isolated=isolated, prefix=prefix)
if root_is_purelib(name, wheeldir):
return scheme['purelib']
else:
return scheme['platlib']
def test_purelib_platlib(self, data):
"""
Test the "wheel is purelib/platlib" code.
"""
packages = [
("pure_wheel", data.packages.join("pure_wheel-1.7"), True),
("plat_wheel", data.packages.join("plat_wheel-1.7"), False),
("pure_wheel", data.packages.join(
"pure_wheel-_invalidversion_"), True),
("plat_wheel", data.packages.join(
"plat_wheel-_invalidversion_"), False),
]
for name, path, expected in packages:
assert wheel.root_is_purelib(name, path) == expected
def test_purelib_platlib(self, data):
"""
Test the "wheel is purelib/platlib" code.
"""
packages = [
("pure_wheel", data.packages.joinpath("pure_wheel-1.7"), True),
("plat_wheel", data.packages.joinpath("plat_wheel-1.7"), False),
("pure_wheel",
data.packages.joinpath("pure_wheel-_invalidversion_"), True),
("plat_wheel",
data.packages.joinpath("plat_wheel-_invalidversion_"), False),
]
for name, path, expected in packages:
assert wheel.root_is_purelib(name, path) == expected
def lib_dir(name,
wheeldir,
user=False,
home=None,
root=None,
isolated=False,
prefix=None):
scheme = distutils_scheme("",
user=user,
home=home,
root=root,
isolated=isolated,
prefix=prefix)
if root_is_purelib(name, wheeldir):
return scheme['purelib']
else:
return scheme['platlib']
