add unit test for get_wheel_architecture

Author: mayeut

tests/unit/test_wheeltools.py

@@ -0,0 +1,43 @@
+from __future__ import annotations
+
+import re
+
+import pytest
+
+from auditwheel.architecture import Architecture
+from auditwheel.error import NonPlatformWheel
+from auditwheel.wheeltools import WheelToolsError, get_wheel_architecture
+
+
+@pytest.mark.parametrize(
+    ("filename", "expected"),
+    [(f"foo-1.0-py3-none-linux_{arch}.whl", arch) for arch in Architecture]
+    + [("foo-1.0-py3-none-linux_x86_64.manylinux1_x86_64.whl", Architecture.x86_64)],
+)
+def test_get_wheel_architecture(filename: str, expected: Architecture) -> None:
+    arch = get_wheel_architecture(filename)
+    assert arch == expected.baseline
+
+
+def test_get_wheel_architecture_unknown() -> None:
+    with pytest.raises(WheelToolsError, match=re.escape("unknown architecture")):
+        get_wheel_architecture("foo-1.0-py3-none-linux_mipsel.whl")
+
+
+def test_get_wheel_architecture_pure() -> None:
+    with pytest.raises(NonPlatformWheel):
+        get_wheel_architecture("foo-1.0-py3-none-any.whl")
+
+
+@pytest.mark.parametrize(
+    "filename",
+    [
+        "foo-1.0-py3-none-linux_x86_64.linux_aarch64.whl",
+        "foo-1.0-py3-none-linux_x86_64.linux_mipsel.whl",
+        "foo-1.0-py3-none-linux_x86_64.any.whl",
+    ],
+)
+def test_get_wheel_architecture_multiple(filename: str) -> None:
+    match = re.escape("multiple architectures are not supported")
+    with pytest.raises(WheelToolsError, match=match):
+        get_wheel_architecture(filename)