我正在使用 Fortran 创建一个 Python 模块f2py
。如果在 Fortran 模块中遇到错误,我想在 Python 程序中产生错误(包括错误消息)。考虑以下示例:
Fortran 代码(test.f):
subroutine foo(a,m)
integer :: m,i
integer, dimension(m) :: a
!f2py intent(in) :: m
!f2py intent(in,out) :: a
!f2py intent(hide), depend(a) :: m=shape(a)
do i = 1,m
if ( a(i) .eq. 0 ) then
print*, 'ERROR HERE..?'
end if
a(i) = a(i)+1
end do
end subroutine
这个非常简单的程序添加1
到a
. a(i)
但如果等于 0,应该会产生错误。随附的 Python 代码:
import test
print test.foo(np.array([1,2],dtype='uint32'))
print test.foo(np.array([0,2],dtype='uint32'))
现在的输出是:
[2 3]
ERROR HERE..?
[1 3]
但我希望 Python 程序保留错误。请帮忙。
回答
stop
Fortran 中的命令正是这样做的。考虑更新的 Fortran 代码:
subroutine foo(a,m)
integer :: m,i
integer, dimension(m) :: a
!f2py intent(in) :: m
!f2py intent(in,out) :: a
!f2py intent(hide), depend(a) :: m=shape(a)
do i = 1,m
if ( a(i) .eq. 0 ) then
print*, 'Error from Fortran'
stop
end if
a(i) = a(i)+1
end do
end subroutine
现在的输出是:
[2 3]
Error from Fortran
即 Python 代码在出错后不会继续。