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
|
__usage__ = """
Run:
python return_integer.py [<f2py options>]
Examples:
python return_integer.py --fcompiler=Gnu --no-wrap-functions
python return_integer.py --quiet
"""
import numpy.f2py as f2py2e
from numpy.base import array
def build(f2py_opts):
try:
import f77_ext_return_integer
except ImportError:
assert not f2py2e.compile('''\
function t0(value)
integer value
integer t0
t0 = value
end
function t1(value)
integer*1 value
integer*1 t1
t1 = value
end
function t2(value)
integer*2 value
integer*2 t2
t2 = value
end
function t4(value)
integer*4 value
integer*4 t4
t4 = value
end
function t8(value)
integer*8 value
integer*8 t8
t8 = value
end
subroutine s0(t0,value)
integer value
integer t0
cf2py intent(out) t0
t0 = value
end
subroutine s1(t1,value)
integer*1 value
integer*1 t1
cf2py intent(out) t1
t1 = value
end
subroutine s2(t2,value)
integer*2 value
integer*2 t2
cf2py intent(out) t2
t2 = value
end
subroutine s4(t4,value)
integer*4 value
integer*4 t4
cf2py intent(out) t4
t4 = value
end
subroutine s8(t8,value)
integer*8 value
integer*8 t8
cf2py intent(out) t8
t8 = value
end
''','f77_ext_return_integer',f2py_opts,source_fn='f77_ret_int.f')
from f77_ext_return_integer import t0,t1,t2,t4,t8,s0,s1,s2,s4,s8
test_functions = [t0,t1,t2,t4,t8,s0,s1,s2,s4,s8]
return test_functions
def runtest(t):
import sys
assert t(123)==123,`t(123)`
assert t(123.6)==123
assert t(123l)==123
if sys.version[:3]<'2.3':
assert t(123.6+3j)==123
assert t('123')==123
assert t(-123)==-123
assert t([123])==123
assert t((123,))==123
assert t(array(123))==123
assert t(array([123]))==123
assert t(array([[123]]))==123
assert t(array([123],'b'))==123
assert t(array([123],'h'))==123
assert t(array([123],'i'))==123
assert t(array([123],'l'))==123
assert t(array([123],'B'))==123
assert t(array([123],'f'))==123
assert t(array([123],'d'))==123
if sys.version[:3]<'2.3':
assert t(array([123+3j],'F'))==123
assert t(array([123],'D'))==123
try: raise RuntimeError,`t(array([123],'c'))`
except ValueError: pass
try: raise RuntimeError,`t('abc')`
except ValueError: pass
try: raise RuntimeError,`t([])`
except IndexError: pass
try: raise RuntimeError,`t(())`
except IndexError: pass
try: raise RuntimeError,`t(t)`
except TypeError: pass
try: raise RuntimeError,`t({})`
except TypeError: pass
if t.__doc__.split()[0] in ['t8','s8']:
try: raise RuntimeError,`t(100000000000000000000000l)`
except OverflowError: pass
try: raise RuntimeError,`t(10000000011111111111111.23)`
except OverflowError: pass
else:
if sys.version[:3]<'2.3':
try: raise RuntimeError,`t(10000000000000l)`
except OverflowError: pass
try: raise RuntimeError,`t(10000000000.23)`
except OverflowError: pass
if __name__=='__main__':
#import libwadpy
status = 1
try:
repeat,f2py_opts = f2py2e.f2py_testing.cmdline()
test_functions = build(f2py_opts)
f2py2e.f2py_testing.run(runtest,test_functions,repeat)
print 'ok'
status = 0
finally:
if status:
print '*'*20
print 'Running f2py2e.diagnose'
import numpy.f2py.diagnose as diagnose
#diagnose.run()
|
