weaviate-python-client/test/test_server_version.py at c3fcb0ba0232771a741770825e52c4109bdc9fda · weaviate/weaviate-python-client · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
import pytest

from weaviate.util import _ServerVersion


@pytest.mark.parametrize(
    "version_string,expected",
    [
        ("1.2.3.a", _ServerVersion(1, 2, 3)),
        ("1.2.3.4", _ServerVersion(1, 2, 3)),
        ("v1.2.3.4", _ServerVersion(1, 2, 3)),
        ("1.2.3", _ServerVersion(1, 2, 3)),
        ("v1.2.3", _ServerVersion(1, 2, 3)),
        ("1.2", _ServerVersion(1, 2, 0)),
        ("v1.2", _ServerVersion(1, 2, 0)),
        ("1", _ServerVersion(1, 0, 0)),
        ("v1", _ServerVersion(1, 0, 0)),
        ("", _ServerVersion(0, 0, 0)),
    ],
)
def test_server_version_successful_parsing(version_string: str, expected: _ServerVersion) -> None:
    assert _ServerVersion.from_string(version_string) == expected


@pytest.mark.parametrize(
    "version_string",
    [
        "v",
        "1.v2.3",
        "1.2.b3",
    ],
)
def test_server_version_unsuccessful_parsing(version_string: str) -> None:
    with pytest.raises(ValueError):
        _ServerVersion.from_string(version_string)


@pytest.mark.parametrize(
    "is_valid",
    [
        _ServerVersion(1, 2, 3).is_at_least(1, 2, 3),
        _ServerVersion(1, 2, 3).is_at_least(1, 2, 2),
        _ServerVersion(1, 2, 3).is_at_least(1, 1, 3),
        _ServerVersion(1, 2, 3).is_at_least(1, 1, 2),
        _ServerVersion(1, 2, 3).is_at_least(0, 2, 3),
        _ServerVersion(1, 2, 3).is_at_least(0, 2, 2),
        _ServerVersion(1, 2, 3).is_at_least(0, 1, 3),
        _ServerVersion(1, 2, 3).is_at_least(0, 1, 2),
        not _ServerVersion(1, 2, 3).is_at_least(1, 2, 4),
        not _ServerVersion(1, 2, 3).is_at_least(1, 3, 3),
        not _ServerVersion(1, 2, 3).is_at_least(1, 3, 4),
        not _ServerVersion(1, 2, 3).is_at_least(2, 2, 3),
        not _ServerVersion(1, 2, 3).is_at_least(2, 2, 4),
        not _ServerVersion(1, 2, 3).is_at_least(2, 3, 3),
        not _ServerVersion(1, 2, 3).is_at_least(2, 3, 4),
        not _ServerVersion(0, 0, 0).is_at_least(0, 0, 1),
    ],
)
def test_server_version_is_at_least(is_valid: bool) -> None:
    assert is_valid


@pytest.mark.parametrize(
    "version,expected",
    [
        ("1.36.9", False),
        ("1.37.14", False),
        ("1.37.15", True),
        ("1.37.99", True),
        ("1.38.0", False),
        ("1.38.7", False),
        ("1.38.8", True),
        ("1.38.20", True),
        ("1.39.0", True),
        ("1.40.0", True),
        ("2.0.0", True),
    ],
)
def test_server_version_is_at_least_any(version: str, expected: bool) -> None:
    assert (
        _ServerVersion.from_string(version).is_at_least_any((1, 37, 15), (1, 38, 8), (1, 39, 0))
        is expected
    )


def test_server_version_is_at_least_any_single_minimum() -> None:
    assert _ServerVersion(1, 39, 0).is_at_least_any((1, 39, 0))
    assert not _ServerVersion(1, 38, 99).is_at_least_any((1, 39, 0))


def test_server_version_magic_methods() -> None:
    # Test __eq__
    assert _ServerVersion(1, 2, 3) == _ServerVersion(1, 2, 3)
    # Test __neq__
    assert _ServerVersion(1, 2, 3) != _ServerVersion(1, 2, 2)

    # Test __lt__
    assert _ServerVersion(1, 2, 2) < _ServerVersion(1, 2, 3)
    assert _ServerVersion(1, 1, 3) < _ServerVersion(1, 2, 3)
    assert _ServerVersion(0, 2, 3) < _ServerVersion(1, 2, 3)

    # Test __le__
    assert _ServerVersion(1, 2, 3) <= _ServerVersion(1, 2, 3)
    assert _ServerVersion(1, 2, 2) <= _ServerVersion(1, 2, 3)
    assert _ServerVersion(1, 1, 3) <= _ServerVersion(1, 2, 3)
    assert _ServerVersion(0, 2, 3) <= _ServerVersion(1, 2, 3)

    # Test __gt__
    assert _ServerVersion(1, 2, 4) > _ServerVersion(1, 2, 3)
    assert _ServerVersion(1, 3, 3) > _ServerVersion(1, 2, 3)
    assert _ServerVersion(2, 2, 3) > _ServerVersion(1, 2, 3)

    # Test __ge__
    assert _ServerVersion(1, 2, 3) >= _ServerVersion(1, 2, 3)
    assert _ServerVersion(1, 2, 4) >= _ServerVersion(1, 2, 3)
    assert _ServerVersion(1, 3, 3) >= _ServerVersion(1, 2, 3)
    assert _ServerVersion(2, 2, 3) >= _ServerVersion(1, 2, 3)