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
|
import pytest
import os
import sys
test_data = [
(b''),
(os.urandom(8 * 1024)),
(b'0' * 8 * 1024),
(bytearray(b'')),
(bytearray(os.urandom(8 * 1024))),
(bytearray(open(os.path.join(os.path.dirname(__file__), 'numpy_byte_array.bin'), 'rb').read()))
]
if sys.version_info > (2, 7):
test_data += [
(memoryview(b'')),
(memoryview(os.urandom(8 * 1024)))
]
@pytest.fixture(
params=test_data,
ids=[
'data' + str(i) for i in range(len(test_data))
]
)
def data(request):
return request.param
@pytest.fixture(
params=[
(
{
'store_size': True
}
),
(
{
'store_size': False
}
),
]
)
def store_size(request):
return request.param
@pytest.fixture(
params=[
(
{
'return_bytearray': True
}
),
(
{
'return_bytearray': False
}
),
]
)
def return_bytearray(request):
return request.param
@pytest.fixture
def c_return_bytearray(return_bytearray):
return return_bytearray
@pytest.fixture
def d_return_bytearray(return_bytearray):
return return_bytearray
@pytest.fixture(
params=[
('fast', None)
] + [
('fast', {'acceleration': s}) for s in range(10)
] + [
('high_compression', None)
] + [
('high_compression', {'compression': s}) for s in range(17)
] + [
(None, None)
]
)
def mode(request):
return request.param
dictionary = [
None,
(0, 0),
(100, 200),
(0, 8 * 1024),
os.urandom(8 * 1024)
]
@pytest.fixture(
params=dictionary,
ids=[
'dictionary' + str(i) for i in range(len(dictionary))
]
)
def dictionary(request):
return request.param
|