diff options
author | Benoit Pierre <benoit.pierre@gmail.com> | 2019-11-25 11:24:10 +0100 |
---|---|---|
committer | Benoit Pierre <benoit.pierre@gmail.com> | 2019-11-25 11:39:11 +0100 |
commit | 926c80f5e84823f48103f3695f55f23949cc5d37 (patch) | |
tree | ea515585d307bad32323f6c34cac91105b1ef37c /setuptools | |
parent | d155aa0d61690b7013de968b912a001d18f5cfdd (diff) | |
download | external_python_setuptools-926c80f5e84823f48103f3695f55f23949cc5d37.tar.gz external_python_setuptools-926c80f5e84823f48103f3695f55f23949cc5d37.tar.bz2 external_python_setuptools-926c80f5e84823f48103f3695f55f23949cc5d37.zip |
wheel: fix `is_compatible` implementation
Diffstat (limited to 'setuptools')
-rw-r--r-- | setuptools/tests/test_wheel.py | 9 | ||||
-rw-r--r-- | setuptools/wheel.py | 2 |
2 files changed, 10 insertions, 1 deletions
diff --git a/setuptools/tests/test_wheel.py b/setuptools/tests/test_wheel.py index d50816c2..55d346c6 100644 --- a/setuptools/tests/test_wheel.py +++ b/setuptools/tests/test_wheel.py @@ -18,6 +18,7 @@ import pytest from pkg_resources import Distribution, PathMetadata, PY_MAJOR from setuptools.extern.packaging.utils import canonicalize_name +from setuptools.extern.packaging.tags import parse_tag from setuptools.wheel import Wheel from .contexts import tempdir @@ -571,3 +572,11 @@ def test_wheel_no_dist_dir(): _check_wheel_install(wheel_path, install_dir, None, project_name, version, None) + + +def test_wheel_is_compatible(monkeypatch): + def sys_tags(): + for t in parse_tag('cp36-cp36m-manylinux1_x86_64'): + yield t + monkeypatch.setattr('setuptools.wheel.sys_tags', sys_tags) + assert Wheel('onnxruntime-0.1.2-cp36-cp36m-manylinux1_x86_64.whl').is_compatible() diff --git a/setuptools/wheel.py b/setuptools/wheel.py index 3effd79b..025aaa82 100644 --- a/setuptools/wheel.py +++ b/setuptools/wheel.py @@ -77,7 +77,7 @@ class Wheel: def is_compatible(self): '''Is the wheel is compatible with the current platform?''' - supported_tags = set(map(str, sys_tags())) + supported_tags = set((t.interpreter, t.abi, t.platform) for t in sys_tags()) return next((True for t in self.tags() if t in supported_tags), False) def egg_name(self): |