diff --git a/tests/test_version.py b/tests/test_version.py index 4d3f905..38e74ec 100644 --- a/tests/test_version.py +++ b/tests/test_version.py @@ -5,9 +5,11 @@ import unittest -from openwrt_luci_rpc.utilities import is_legacy_version + from packaging import version +from openwrt_luci_rpc.utilities import is_legacy_version + class TestOpenwrtLuciRPC(unittest.TestCase): """Tests for `openwrt_luci_rpc` package.""" @@ -21,8 +23,8 @@ def tearDown(self): def test_is_legacy_version(self): """Test comparing versions works as expected.""" - assert is_legacy_version(version.parse("15.05")) is True - assert is_legacy_version(version.parse("15.05.1")) is True + assert is_legacy_version(version.parse("15.05")) is False + assert is_legacy_version(version.parse("15.05.1")) is False assert is_legacy_version(version.parse("17.01")) is True assert is_legacy_version(version.parse("17.01.6")) is True