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
|
from typing import Any
import pytest
from skytools.tnetstrings import dumps, loads
def ustr(v: Any) -> str:
return repr(v).replace("u'", "'")
def nstr(b: bytes) -> str:
return b.decode('utf8')
def test_dumps_simple_values() -> None:
vals = (None, False, True, 222, 333.0, b"foo", u"bar")
tnvals = [nstr(dumps(v)) for v in vals]
assert tnvals == [
'0:~', '5:false!', '4:true!', '3:222#', '5:333.0^', '3:foo,', '3:bar,'
]
def test_dumps_complex_values() -> None:
vals2 = [[], (), {}, [1, 2], {'a': 'b'}]
tnvals2 = [nstr(dumps(v)) for v in vals2]
assert tnvals2 == ['0:]', '0:]', '0:}', '8:1:1#1:2#]', '8:1:a,1:b,}']
def test_loads_simple() -> None:
vals = (None, False, True, 222, 333.0, b"foo", u"bar")
res = ustr([loads(dumps(v)) for v in vals])
assert res == "[None, False, True, 222, 333.0, 'foo', 'bar']"
def test_loads_complex() -> None:
vals2 = [[], (), {}, [1, 2], {'a': 'b'}]
res = ustr([loads(dumps(v)) for v in vals2])
assert res == "[[], [], {}, [1, 2], {'a': 'b'}]"
def test_dumps_mview() -> None:
res = nstr(dumps([memoryview(b'zzz'), b'qqq']))
assert res == '12:3:zzz,3:qqq,]'
def test_loads_errors() -> None:
with pytest.raises(ValueError):
loads(b'4:qwez!')
with pytest.raises(ValueError):
loads(b'4:')
with pytest.raises(ValueError):
loads(b'4:qwez')
with pytest.raises(ValueError):
loads(b'4')
with pytest.raises(ValueError):
loads(b'')
with pytest.raises(ValueError):
loads(b'999999999999999999:z,')
with pytest.raises(TypeError):
loads(u'qweqwe') # type: ignore[arg-type]
with pytest.raises(ValueError):
loads(b'4:true!0:~')
with pytest.raises(ValueError):
loads(b'1:X~')
with pytest.raises(ValueError):
loads(b'8:1:1#1:2#}')
with pytest.raises(ValueError):
loads(b'1:Xz')
def test_dumps_errors() -> None:
with pytest.raises(TypeError):
dumps(open)
|
