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
|
__usage__ = """
Run:
python return_character.py [<f2py options>]
Examples:
python return_character.py --fcompiler=Gnu --no-wrap-functions
python return_character.py --quiet
"""
import f2py2e
from Numeric import array
def build(f2py_opts):
try:
import f90_ext_return_character
except ImportError:
assert not f2py2e.compile('''\
module f90_return_char
contains
function t0(value)
character :: value
character :: t0
t0 = value
end function t0
function t1(value)
character(len=1) :: value
character(len=1) :: t1
t1 = value
end function t1
function t5(value)
character(len=5) :: value
character(len=5) :: t5
t5 = value
end function t5
function ts(value)
character(len=*) :: value
character(len=10) :: ts
ts = value
end function ts
subroutine s0(t0,value)
character :: value
character :: t0
!f2py intent(out) t0
t0 = value
end subroutine s0
subroutine s1(t1,value)
character(len=1) :: value
character(len=1) :: t1
!f2py intent(out) t1
t1 = value
end subroutine s1
subroutine s5(t5,value)
character(len=5) :: value
character(len=5) :: t5
!f2py intent(out) t5
t5 = value
end subroutine s5
subroutine ss(ts,value)
character(len=*) :: value
character(len=10) :: ts
!f2py intent(out) ts
ts = value
end subroutine ss
end module f90_return_char
''','f90_ext_return_character',f2py_opts,source_fn='f90_ret_char.f90')
from f90_ext_return_character import f90_return_char as m
test_functions = [m.t0,m.t1,m.t5,m.ts,m.s0,m.s1,m.s5,m.ss]
return test_functions
def runtest(t):
tname = t.__doc__.split()[0]
if tname in ['t0','t1','s0','s1']:
assert t(23)=='2'
r = t('ab');assert r=='a',`r`
r = t(array('ab'));assert r=='a',`r`
r = t(array(77,'1'));assert r=='M',`r`
try: raise RuntimeError,`t(array([77,87]))`
except ValueError: pass
try: raise RuntimeError,`t(array(77))`
except ValueError: pass
elif tname in ['ts','ss']:
assert t(23)=='23 ',`t(23)`
assert t('123456789abcdef')=='123456789a',`t('123456789abcdef')`
elif tname in ['t5','s5']:
assert t(23)=='23 '
assert t('ab')=='ab '
assert t('123456789abcdef')=='12345'
else:
raise NotImplementedError
if __name__=='__main__':
#import libwadpy
repeat,f2py_opts = f2py2e.f2py_testing.cmdline()
test_functions = build(f2py_opts)
f2py2e.f2py_testing.run(runtest,test_functions,repeat)
print 'ok'
|
