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
|
""" Testing
"""
import os
import sys
if sys.version_info[0] >= 3:
from io import BytesIO
cStringIO = BytesIO
else:
from cStringIO import StringIO as cStringIO
from StringIO import StringIO as BytesIO
from tempfile import mkstemp
import numpy as np
from numpy.compat import asbytes
from nose.tools import assert_true, assert_false, \
assert_equal, assert_raises
from numpy.testing import assert_array_equal, assert_array_almost_equal, \
run_module_suite
from scipy.io.matlab.streams import make_stream, \
GenericStream, cStringStream, FileStream, \
_read_into, _read_string
def setup():
val = asbytes('a\x00string')
global fs, gs, cs, fname
fd, fname = mkstemp()
fs = os.fdopen(fd, 'wb')
fs.write(val)
fs.close()
fs = open(fname, 'rb')
gs = BytesIO(val)
cs = cStringIO(val)
def teardown():
global fname, fs
fs.close()
del fs
os.unlink(fname)
def test_make_stream():
global fs, gs, cs
# test stream initialization
assert_true(isinstance(make_stream(gs), GenericStream))
if sys.version_info[0] < 3:
assert_true(isinstance(make_stream(cs), cStringStream))
assert_true(isinstance(make_stream(fs), FileStream))
def test_tell_seek():
global fs, gs, cs
for s in (fs, gs, cs):
st = make_stream(s)
res = st.seek(0)
yield assert_equal, res, 0
yield assert_equal, st.tell(), 0
res = st.seek(5)
yield assert_equal, res, 0
yield assert_equal, st.tell(), 5
res = st.seek(2, 1)
yield assert_equal, res, 0
yield assert_equal, st.tell(), 7
res = st.seek(-2, 2)
yield assert_equal, res, 0
yield assert_equal, st.tell(), 6
def test_read():
global fs, gs, cs
for s in (fs, gs, cs):
st = make_stream(s)
st.seek(0)
res = st.read(-1)
yield assert_equal, res, asbytes('a\x00string')
st.seek(0)
res = st.read(4)
yield assert_equal, res, asbytes('a\x00st')
# read into
st.seek(0)
res = _read_into(st, 4)
yield assert_equal, res, asbytes('a\x00st')
res = _read_into(st, 4)
yield assert_equal, res, asbytes('ring')
yield assert_raises, IOError, _read_into, st, 2
# read alloc
st.seek(0)
res = _read_string(st, 4)
yield assert_equal, res, asbytes('a\x00st')
res = _read_string(st, 4)
yield assert_equal, res, asbytes('ring')
yield assert_raises, IOError, _read_string, st, 2
if __name__ == "__main__":
run_module_suite()
|
