|
|
"""Tests for distutils.version.""" |
|
|
|
|
|
import distutils |
|
|
from distutils.version import LooseVersion, StrictVersion |
|
|
|
|
|
import pytest |
|
|
|
|
|
|
|
|
@pytest.fixture(autouse=True) |
|
|
def suppress_deprecation(): |
|
|
with distutils.version.suppress_known_deprecation(): |
|
|
yield |
|
|
|
|
|
|
|
|
class TestVersion: |
|
|
def test_prerelease(self): |
|
|
version = StrictVersion('1.2.3a1') |
|
|
assert version.version == (1, 2, 3) |
|
|
assert version.prerelease == ('a', 1) |
|
|
assert str(version) == '1.2.3a1' |
|
|
|
|
|
version = StrictVersion('1.2.0') |
|
|
assert str(version) == '1.2' |
|
|
|
|
|
def test_cmp_strict(self): |
|
|
versions = ( |
|
|
('1.5.1', '1.5.2b2', -1), |
|
|
('161', '3.10a', ValueError), |
|
|
('8.02', '8.02', 0), |
|
|
('3.4j', '1996.07.12', ValueError), |
|
|
('3.2.pl0', '3.1.1.6', ValueError), |
|
|
('2g6', '11g', ValueError), |
|
|
('0.9', '2.2', -1), |
|
|
('1.2.1', '1.2', 1), |
|
|
('1.1', '1.2.2', -1), |
|
|
('1.2', '1.1', 1), |
|
|
('1.2.1', '1.2.2', -1), |
|
|
('1.2.2', '1.2', 1), |
|
|
('1.2', '1.2.2', -1), |
|
|
('0.4.0', '0.4', 0), |
|
|
('1.13++', '5.5.kw', ValueError), |
|
|
) |
|
|
|
|
|
for v1, v2, wanted in versions: |
|
|
try: |
|
|
res = StrictVersion(v1)._cmp(StrictVersion(v2)) |
|
|
except ValueError: |
|
|
if wanted is ValueError: |
|
|
continue |
|
|
else: |
|
|
raise AssertionError(f"cmp({v1}, {v2}) shouldn't raise ValueError") |
|
|
assert res == wanted, f'cmp({v1}, {v2}) should be {wanted}, got {res}' |
|
|
res = StrictVersion(v1)._cmp(v2) |
|
|
assert res == wanted, f'cmp({v1}, {v2}) should be {wanted}, got {res}' |
|
|
res = StrictVersion(v1)._cmp(object()) |
|
|
assert res is NotImplemented, ( |
|
|
f'cmp({v1}, {v2}) should be NotImplemented, got {res}' |
|
|
) |
|
|
|
|
|
def test_cmp(self): |
|
|
versions = ( |
|
|
('1.5.1', '1.5.2b2', -1), |
|
|
('161', '3.10a', 1), |
|
|
('8.02', '8.02', 0), |
|
|
('3.4j', '1996.07.12', -1), |
|
|
('3.2.pl0', '3.1.1.6', 1), |
|
|
('2g6', '11g', -1), |
|
|
('0.960923', '2.2beta29', -1), |
|
|
('1.13++', '5.5.kw', -1), |
|
|
) |
|
|
|
|
|
for v1, v2, wanted in versions: |
|
|
res = LooseVersion(v1)._cmp(LooseVersion(v2)) |
|
|
assert res == wanted, f'cmp({v1}, {v2}) should be {wanted}, got {res}' |
|
|
res = LooseVersion(v1)._cmp(v2) |
|
|
assert res == wanted, f'cmp({v1}, {v2}) should be {wanted}, got {res}' |
|
|
res = LooseVersion(v1)._cmp(object()) |
|
|
assert res is NotImplemented, ( |
|
|
f'cmp({v1}, {v2}) should be NotImplemented, got {res}' |
|
|
) |
|
|
|