MMKV/Python/unit_test.py at master · Tencent/MMKV · 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
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
#!/usr/bin/env python
# -*- coding: utf-8 -*-

import sys
import mmkv
import tempfile

KeyNotExist = 'KeyNotExist'


def test_bool(kv):
    ret = kv.set(True, 'bool')
    assert ret

    value = kv.getBool('bool')
    assert value

    value = kv.getBool(KeyNotExist)
    assert not value

    value = kv.getBool(KeyNotExist, True)
    assert value

    print('test bool: passed')


def test_int(kv):
    MinInt = -1 * (2 ** 31)
    ret = kv.set(MinInt, 'Int')
    assert ret

    value = kv.getInt('Int')
    assert value == MinInt

    MaxInt = (2 ** 31) - 1
    ret = kv.set(MaxInt, 'Int')
    assert ret

    value = kv.getInt('Int')
    assert value == MaxInt

    value = kv.getInt(KeyNotExist)
    assert value == 0

    value = kv.getInt(KeyNotExist, -1)
    assert value == -1

    print('test int: passed')


def test_uint(kv):
    MaxUInt = (2 ** 32) - 1
    ret = kv.set(MaxUInt, 'UInt')
    assert ret

    value = kv.getUInt('UInt')
    assert value == MaxUInt

    value = kv.getUInt(KeyNotExist)
    assert value == 0

    value = kv.getUInt(KeyNotExist, 1)
    assert value == 1

    print('test unsigned int: passed')


def test_long_int(kv):
    MinLongInt = -1 * (2 ** 63)
    ret = kv.set(MinLongInt, 'LongInt')
    assert ret

    value = kv.getLongInt('LongInt')
    assert value == MinLongInt

    MaxLongInt = (2 ** 63) - 1
    ret = kv.set(MaxLongInt, 'LongInt')
    assert ret

    value = kv.getLongInt('LongInt')
    assert value == MaxLongInt

    value = kv.getLongInt(KeyNotExist)
    assert value == 0

    value = kv.getLongInt(KeyNotExist, -1)
    assert value == -1

    print('test long int: passed')


def test_long_uint(kv):
    MaxLongUInt = (2 ** 64) - 1
    ret = kv.set(MaxLongUInt, 'LongUInt')
    assert ret

    value = kv.getLongUInt('LongUInt')
    assert value == MaxLongUInt

    value = kv.getLongUInt(KeyNotExist)
    assert value == 0

    value = kv.getLongUInt(KeyNotExist, 1)
    assert value == 1

    print('test long unsigned int: passed')


def is_float_equal(a, b, accuracy=1e-09):
    return abs(a - b) <= max(accuracy * max(abs(a), abs(b)), 0.0)


def test_float(kv):
    MinFloat = sys.float_info.min
    ret = kv.set(MinFloat, 'Float')
    assert ret

    value = kv.getFloat('Float')
    assert is_float_equal(value, MinFloat)

    MaxFloat = sys.float_info.max
    ret = kv.set(MaxFloat, 'Float')
    assert ret

    value = kv.getFloat('Float')
    assert is_float_equal(value, MaxFloat)

    value = kv.getFloat(KeyNotExist)
    assert is_float_equal(value, 0.0)

    value = kv.getFloat(KeyNotExist, -1.0)
    assert is_float_equal(value, -1.0)

    print('test float: passed')


def test_string(kv):
    OlympicString = 'Hello 2021 Olympic Games 奥运会'
    ret = kv.set(OlympicString, 'String')
    assert ret

    value = kv.getString('String')
    if sys.version_info >= (3, 0):
        assert value == OlympicString
    else:
        assert value == OlympicString.decode('utf-8')

    value = kv.getString(KeyNotExist)
    assert (not value) or (len(value) == 0)

    value = kv.getString(KeyNotExist, 'a')
    assert value == 'a'

    print('test string: passed')


def test_bytes(kv):
    OlympicYears = [0, 4, 8, 12, 16, 21]
    OlympicYearsBytes = bytes(OlympicYears)
    ret = kv.set(OlympicYearsBytes, 'Bytes')
    assert ret

    value = kv.getBytes('Bytes')
    assert value == OlympicYearsBytes
    if sys.version_info >= (3, 0):
        value = list(value)
        assert value == OlympicYears

    value = kv.getBytes(KeyNotExist)
    assert (not value) or (len(value) == 0)

    default_value = None
    if sys.version_info >= (3, 0):
        default_value = bytes('哦豁', 'utf8')
    else:
        default_value = bytes('a')
    value = kv.getBytes(KeyNotExist, default_value)
    assert value == default_value

    print('test bytes: passed')


def test_equal(kv, mmap_id):
    assert kv.mmapID() == mmap_id

    another = mmkv.MMKV(mmap_id)
    assert kv == another

    kv_keys = sorted(kv.keys())
    another_keys = sorted(another.keys())
    assert kv_keys == another_keys

    print('test equal: passed')


if __name__ == '__main__':
    temp_dir = tempfile.gettempdir()
    root_dir = temp_dir + '/mmkv'
    mmkv.MMKV.initializeMMKV(root_dir)
    kv = mmkv.MMKV('unit_test_python')
    test_bool(kv)
    test_int(kv)
    test_uint(kv)
    test_long_int(kv)
    test_long_uint(kv)
    test_float(kv)
    test_string(kv)
    test_bytes(kv)
    test_equal(kv, 'unit_test_python')