11
11
# - array type only supports old buffer interface
12
12
# - array.frombytes is not available, must use deprecated array.fromstring
13
13
if sys .version_info [0 ] < 3 :
14
- def __memoryview (obj ):
14
+ def make_memoryview (obj ):
15
15
return memoryview (buffer (obj ))
16
16
17
- def __make_array (f , data ):
17
+ def make_array (f , data ):
18
18
a = array (f )
19
19
a .fromstring (data )
20
20
return a
21
21
22
- def __get_data (a ):
22
+ def get_data (a ):
23
23
return a .tostring ()
24
24
else :
25
- __memoryview = memoryview
25
+ make_memoryview = memoryview
26
26
27
- def __make_array (f , data ):
27
+ def make_array (f , data ):
28
28
a = array (f )
29
29
a .frombytes (data )
30
30
return a
31
31
32
- def __get_data (a ):
32
+ def get_data (a ):
33
33
return a .tobytes ()
34
34
35
35
36
36
def __run_test (format , nbytes , expected_header , expected_prefix , use_bin_type ):
37
37
# create a new array
38
38
original_array = array (format )
39
39
original_array .fromlist ([255 ] * (nbytes // original_array .itemsize ))
40
- original_data = __get_data (original_array )
41
- view = __memoryview (original_array )
40
+ original_data = get_data (original_array )
41
+ view = make_memoryview (original_array )
42
42
43
43
# pack, unpack, and reconstruct array
44
44
packed = packb (view , use_bin_type = use_bin_type )
45
45
unpacked = unpackb (packed )
46
- reconstructed_array = __make_array (format , unpacked )
46
+ reconstructed_array = make_array (format , unpacked )
47
47
48
48
# check that we got the right amount of data
49
49
assert len (original_data ) == nbytes
@@ -57,79 +57,49 @@ def __run_test(format, nbytes, expected_header, expected_prefix, use_bin_type):
57
57
assert original_array == reconstructed_array
58
58
59
59
60
- # -----------
61
- # test fixstr
62
- # -----------
63
-
64
-
65
- def test_memoryview_byte_fixstr ():
60
+ def test_fixstr_from_byte ():
66
61
__run_test ('B' , 31 , b'\xbf ' , b'' , False )
67
62
68
63
69
- def test_memoryview_float_fixstr ():
64
+ def test_fixstr_from_float ():
70
65
__run_test ('f' , 28 , b'\xbc ' , b'' , False )
71
66
72
67
73
- # ----------
74
- # test str16
75
- # ----------
76
-
77
-
78
- def test_memoryview_byte_str16 ():
68
+ def test_str16_from_byte ():
79
69
__run_test ('B' , 2 ** 8 , b'\xda ' , b'\x01 \x00 ' , False )
80
70
81
71
82
- def test_memoryview_float_str16 ():
72
+ def test_str16_from_float ():
83
73
__run_test ('f' , 2 ** 8 , b'\xda ' , b'\x01 \x00 ' , False )
84
74
85
75
86
- # ----------
87
- # test str32
88
- # ----------
89
-
90
-
91
- def test_memoryview_byte_str32 ():
76
+ def test_str32_from_byte ():
92
77
__run_test ('B' , 2 ** 16 , b'\xdb ' , b'\x00 \x01 \x00 \x00 ' , False )
93
78
94
79
95
- def test_memoryview_float_str32 ():
80
+ def test_str32_from_float ():
96
81
__run_test ('f' , 2 ** 16 , b'\xdb ' , b'\x00 \x01 \x00 \x00 ' , False )
97
82
98
83
99
- # ---------
100
- # test bin8
101
- # ---------
102
-
103
-
104
- def test_memoryview_byte_bin8 ():
84
+ def test_bin8_from_byte ():
105
85
__run_test ('B' , 1 , b'\xc4 ' , b'\x01 ' , True )
106
86
107
87
108
- def test_memoryview_float_bin8 ():
88
+ def test_bin8_from_float ():
109
89
__run_test ('f' , 4 , b'\xc4 ' , b'\x04 ' , True )
110
90
111
91
112
- # ----------
113
- # test bin16
114
- # ----------
115
-
116
-
117
- def test_memoryview_byte_bin16 ():
92
+ def test_bin16_from_byte ():
118
93
__run_test ('B' , 2 ** 8 , b'\xc5 ' , b'\x01 \x00 ' , True )
119
94
120
95
121
- def test_memoryview_float_bin16 ():
96
+ def test_bin16_from_float ():
122
97
__run_test ('f' , 2 ** 8 , b'\xc5 ' , b'\x01 \x00 ' , True )
123
98
124
99
125
- # ----------
126
- # test bin32
127
- # ----------
128
-
129
-
130
- def test_memoryview_byte_bin32 ():
100
+ def test_bin32_from_byte ():
131
101
__run_test ('B' , 2 ** 16 , b'\xc6 ' , b'\x00 \x01 \x00 \x00 ' , True )
132
102
133
103
134
- def test_memoryview_float_bin32 ():
104
+ def test_bin32_from_float ():
135
105
__run_test ('f' , 2 ** 16 , b'\xc6 ' , b'\x00 \x01 \x00 \x00 ' , True )
0 commit comments